NumPy提供了一种提取多维数组的行/列的强大方法。例如,考虑我们上面定义的my_array
的例子。
[[4 5] [6 1]]
假设,我们想从中提取第二列(索引1)的所有元素。在这里,我们肉眼可以看出,第二列由两个元素组成:5
和 1
。为此,我们可以执行以下操作:
my_array_column_2 = my_array[:, 1]
print my_array_column_2
注意,我们使用了冒号(:
)而不是行号,而对于列号,我们使用了值1
,最终输出是:[5, 1]
。
我们可以类似地从多维NumPy数组中提取一行。
多维数组切片示例示例图片