如何从 numpy 数组中获取特定列（带有示例）

您可以使用以下语法从 NumPy 数组中获取特定列：

 #get column in index position 2 from NumPy array
my_array[:, 2]

以下示例展示了如何在实践中使用此语法。

示例 1：从 NumPy 数组中获取一列

以下代码显示如何从 NumPy 数组中获取特定列：

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get column in index position 2
data[:, 2]

array([3,7,11])

如果要从 NumPy 数组中获取列并将其作为列向量检索，可以使用以下语法：

 #get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [7],
       [11]])

示例 2：从 NumPy 数组中获取多列

以下代码显示了如何从 NumPy 数组中获取多列：

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2, 4],
       [6, 8],
       [10, 12]])

示例 3：从 NumPy 数组获取范围内的列

以下代码显示如何从 NumPy 数组中获取某个范围内的列：

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

array([[ 1, 2, 3],
       [5, 6, 7],
       [9, 10, 11]])

请注意，范围中的最后一个值（在本例中为 3）不包含在返回的列范围中。

其他资源

以下教程解释了如何在 NumPy 中执行其他常见操作：

如何将函数映射到 NumPy 数组
如何向 NumPy 数组添加列