我有一个二维数组:
a = ([[False False False False False True True True True True True True
True True True True True True True True True True True True
True False False False]
[False False False False True True True True True True True True
True True True True True True True True True True True True
False False False False]])
我试图获取每行中最后一次出现“True”的索引。
所以结果数组应该是
b = ([24, 23])
为了找到第一个 True 的出现,我知道我可以使用 np.argmax()。
b = np.argmax(a==True,axis=1)
有没有从最后查找的函数?我尝试反转数组的值,然后使用 np.argmax(),但它会给出反转数组的索引。
向后处理每一行并减去
行长度 - 1 的结果:
result = a.shape[1] - np.argmax(a[:, ::-1], axis=1) - 1
Even == True
不需要。
结果是:
array([24, 23], dtype=int64)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)