numpy 常用总结

查找a 中 某元素出现的位置索引

a = np.array([1,2,3,4,1,1])

n = np.array([1])

c = np.where(a == 1)

print(c)

result >> (array([0, 4, 5], dtype=int64),)

一个array 索引另外一个array

a = np.array([1,2,3,4,1,1])

n = np.array([0,4,5])

c = a[n]

print(c)

result >>[1 1 1]

在for循环中,range外嵌入enumerate什么意思?

i的array, 存的是j的array的索引