NumPy 高级索引

参考菜鸟教程

1. 一维数组索引

import numpy as np
a = np.array([1,2,3])
b = np.array([1,2,0,0,1,2,0])
c = a[b]
print('c',c.shape,c)

输出

c (7,) [2 3 1 1 2 3 1]

这种索引方式在python中会报错

a[b]等价于遍历b中的元素，作为索引值，取a中索引位置的值。结果与b的长度相同

c = []
for i in b:
    c.append(a[i])

2. 二维数组索引

3. ...和:索引