numpy: как назначить 1d массив путем фильтрации 2d массива?

Question

Скажите, у меня есть следующий код:

import numpy as np

labels = np.array(["A", "B", "C"])

matrix = np.array([[True, False, False], [False, True, True], [False, False, True]])

Как я могу получить 2d массив с метками, где бы он был "True" в матрице? Возможно только с numpy операциями? Таким образом, желаемый результат:

[["A"], ["B", "C"], ["C"]]

Answer 1

Используя понимание списка:

[labels[m].tolist() for m in matrix]

и я предполагаю, что по 2D-массиву вы имеете в виду:

np.array([labels[m].tolist() for m in matrix])