花式索引的索引值是一个数组,对于使用一维整型数组作为索引,如果被索引数据是一维数组,那么索引的结果就是对应位置的元素;如果被索引数据是二维数组,那么索引的结果就是对应下标的行。
对于二维被索引数据来说,索引值可以是二维数据,当索引值为两个维度相等的一维数组组成的二维数组时,以两个维度作为横纵坐标索引出单值后组合成新的一维数组。
import numpy as np x = np.array([[1, 2], [3, 4], [5, 6]]) print('前两行元素为:\n', x[[0, 1]]) print('x[0][0] 和 x[1][1]为:', x[[0, 1], [0, 1]]) print('以下两种格式是一样的:') print(x[[0, 1]][:, [0, 1]]) print(x[0:2, 0:2])
前两行元素为:
[[1 2]
[3 4]]
x[0][0] 和 x[1][1]为: [1 4]
以下两种格式是一样的:
[[1 2]
[3 4]]
[[1 2]
[3 4]]