numpy ndarray 返回 index 问题

Posted 2020-09-28

经常遇到需要返回满足条件的index。 python中没有which函数，但有列表推导式， 可以实现类似功能

y= np.array([3,2,5,20])

y
Out[131]: array([ 3, 2, 5, 20])

[x for x in range(y.shape[0]) if y[x]>3]
Out[129]: [2, 3]    输出满足条件的索引号

 

In [23]: {i:el for i,el in enumerate(["one","two","three"])}
Out[23]: {0: ‘one‘, 1: ‘two‘, 2: ‘three‘}