python神奇索引

神奇索引时Numpy中的术语，用于描述使用整数数组进行数据索引 

arr = np.arange(32).reshape((8, 4))
arr

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19],
       [20, 21, 22, 23],
       [24, 25, 26, 27],
       [28, 29, 30, 31]])

为了选出一个符合特定顺序的子集，可以简单的通过传递一个包含所需顺序的列表或者数组来完成

按顺序选出arr的第5，4，1，7行数据：

arr[[4, 3, 0, 6]]

array([[16, 17, 18, 19],
       [12, 13, 14, 15],
       [ 0,  1,  2,  3],
       [24, 25, 26, 27]])

如果使用负的索引，将从尾部进行选择：

arr[[-3, -5, -7]]

array([[20, 21, 22, 23],
       [12, 13, 14, 15],
       [ 4,  5,  6,  7]])

 传递多个索引数组单独情况有所不同。这样会根据每个索引元组对应的元素选出一个一维数组

arr[[1, 5, 7, 2], [0, 3, 1, 2]]

[1, 5, 7, 2]和[0, 3, 1, 2]对应元素组成索引元组(1, 0),(5, 3),(7, 1),(2, 2)

元素(1, 0),(5, 3),(7, 1),(2, 2)被选中

arr[[1, 5, 7, 2], [0, 3, 1, 2]]

array([ 4, 23, 29, 10])

 选出arr的第2，6，8，3行数据，再按第1，4，2，3列的顺序排列

arr[[1, 5, 7, 2]][:, [0, 3, 1, 2]]

array([[ 4,  7,  5,  6],
       [20, 23, 21, 22],
       [28, 31, 29, 30],
       [ 8, 11,  9, 10]])