如何从 NumPy 中的多维数组中提取特定列？

从 NumPy 中的多维数组中检索列

在 NumPy 中，基本操作是访问多维数组中的单个元素或数据子集。虽然使用语法 test[i] 检索行很简单，但提取特定列可能略有不同。

要访问 NumPy 数组的第 i 列，您可以使用语法 test[:, i]。这将返回一个一维数组，其中包含原始矩阵第 i 列的值。

例如，考虑以下数组：

test = np.array([[1, 2], [3, 4], [5, 6]])

要获取第一列（包含值 1、3 和 5），您可以使用：

>>> test[:, 0]
array([1, 3, 5])

同样，要检索第二列（包含值 2、4 和 6），您可以使用：

>>> test[:, 1]
array([2, 4, 6])

请注意，此操作的计算成本并不是特别昂贵，并且是 NumPy 中处理多维数据的常用技术。

以上是如何从 NumPy 中的多维数组中提取特定列？的详细内容。更多信息请关注PHP中文网其他相关文章！