获取 NumPy 数组的列

本教程将介绍从 Python 的多维 NumPy 数组中获取特定列的方法。

使用基本切片方法从多维 NumPy 数组获取列

基本切片方法的工作原理与 Python 中的列表切片相同。我们可以使用基本的切片方法从多维 NumPy 数组中获取特定的列。基本切片方法将创建现有数组的新视图，而不是创建数组的新副本。这些新的视图指向现有的数组，在所有指向原始数组的视图被垃圾收集器收集之前，原始数组的内存不能被释放。下面的代码例子向我们展示了如何用 Python 的基本切片方法从一个多维 NumPy 数组中获得一个特定的列。

import numpy as np
array = np.array([[1,2],[3,4],[5,6],[7,8],[9,0]])
print(array[:,1])

输出：

[2 4 6 8 0]

在上面的代码中，我们在 Python 中使用 [:,1] 切片索引提取了多维 NumPy 数组 array 的第二列。索引的第一部分是行的索引。我们将第一部分留空，因为我们要选择所有行。:运算符表示索引中的选择操作。我们使用 [:,1] 作为列索引，以获取每一行的第二列。