Как получить два типа индексов одновременно по numpy?

Question

Предположим, у вас есть ndarray z, и вы хотите получить следующие индексы

z_idxes1 = np.argwhere(z == i)
z_idxes2 = np.argwhere(z != i)

Можем ли мы сделать это одним вызовом какой-нибудь простой функции, а не двумя вызовами argwhere?

Answer 1

Не в одном вызове, но вы можете использовать логический оператор not, чтобы получить отрицание вашего массива масок.

mask = (z == i)
z_idxes1 = np.argwhere(mask)

z_idxes2 = np.argwhere(~mask)