python - Numpy 索引第一个 bool 值

Question

在 bool 数组中，我试图获取第一个 True 的列索引。 argmax 至少适用于一个 True，但可以理解 max(False) 为 0。我想知道最好的方法是什么，给定一个非常大的数组。

name = np.array(['a', 'b', 'c', 'd'])
boolarr = np.array([[True, False, False, True],
                    [False, False, True, True],
                    [False, False, False, False]])
colidx = np.argmax(boolarr,axis=1)
print(name[colidx]) #result: ['a', 'c', 'a'] desired: ['a', 'c', None]

Answer 1

您无法更改 argmax 的行为，因为所有 False 的行的最大值是 0。但是，您可以使用 any 来确定包含所有 False 的行，并使用 np.where 来屏蔽你的结果:

out = name[colidx]
np.where(boolarr.any(1), out, None)

array(['a', 'c', None], dtype=object)