Пакет numpy_indexed (отказ от ответственности: я его автор) завершает решение, опубликованное Jaime в приятном и проверенном интерфейсе, а также множество других функций:
import numpy_indexed as npi
new_a = npi.unique(a) # unique elements over axis=0 (rows) by default