假设我有一个numpy数组,如:
a = array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[3, 2, 1]])
我想检查第二个元素== 2.
我知道我可以这样做:
>>> a[:,1]==2
array([ True, False, False, True], dtype=bool)
回归布尔.我的问题是,如何获取条件为真的行的行号？在这个例子中,我想回来,array([0, 3])因为第0和第3行匹配条件第二个元素== 2.