python 二维数组取列

可以使用列表解析式或循环来取二维数组的列。

例如，假设有一个二维数组 `arr`，它有3列4行：

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

要取第二列，可以使用列表解析式：

col = [row[1] for row in arr]

这将返回一个长度为4的列表 `[2,6,10]`，其中包含第二列的所有元素。

如果不使用列表解析式，可以使用循环来实现相同的结果：

col = []
for row in arr:
    col.append(row[1])

这将产生相同的结果。

相关问题

python二维数组取某一列

可以使用列表推导式或numpy库的方法进行操作。假设二维数组为arr，要取第i列，可以使用下面两种方法：

1. 列表推导式

col = [arr[j][i] for j in range(len(arr))]

2. numpy库

import numpy as np

col = np.array(arr)[:, i].tolist()

这两种方法都可以让你得到一个列表，其中存储了二维数组中第i列的所有元素。

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]