经常遇到需要返回满足条件的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‘}
时间: 2024-11-13 06:50:44