我有一个一维的numpy数组。我想得到属性为真的最大和最小索引
比如说,
A = np.array([0, 3, 2, 4, 3, 6, 1, 0])
我想知道A
的值大于或等于4
的最小索引
我能行
i = 0
while A[i] < 4:
i += 1
print("smallest index", i)
i = -1
while A[i] <4:
i -= 1
print("largest index", len(A)+i)
有更好的方法吗
正如this answer中所建议的
np.argmax(A>=4)
返回3
,这确实是最小的索引。但这并没有给我最大的指数
你可以试试这样的东西。根据评论,如果
A
是idx_values
返回满足条件的所有索引值。然后可以访问最小和最大索引位置相关问题 更多 >
编程相关推荐