找到小于x numpy的元素的索引

2024-04-23 22:46:21 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有一个numpy数组,例如:

import numpy as np    
arr = np.array([10,1,2,5,6,2,3,8])

如何提取包含小于6的元素索引的数组,以便得到以下结果:

np.array([1,2,3,5,6])

我想要类似于np.nonzero()的东西,但是它不是测试非零值,而是测试小于x的值

谢谢


Tags: importnumpy元素asnp数组arrayarr