python二维数组选取列

在Python中，可以使用嵌套列表或NumPy库来实现二维数组，下面分别介绍如何选取列：

1. 嵌套列表

假设有一个3行4列的二维数组arr，要选取第2列，则可以使用列表推导式来实现：

arr = [[1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]]

col = [row for row in arr]
print(col)

输出结果为：

[2, 6, 10]

2. NumPy库

如果使用NumPy库创建二维数组，则可以使用切片（slice）来选取列。例如，假设有一个3行4列的二维数组arr，要选取第2列，则可以使用以下代码：

import numpy as np

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

col = arr[:, 1]
print(col)

输出结果为：

[ 2  6 10]