ndarray按列获取

您可以使用NumPy中的索引和切片来按列获取ndarray的值。例如，如果您有一个2维数组arr，您可以使用arr[:, i]来获取第i列的所有值，其中冒号表示所有行。以下是一个示例代码：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 获取第2列
col = arr[:, 1]

print(col)

输出：

[2 5 8]

在这个例子中，我们使用冒号来表示所有行，并使用索引1来表示第2列。