如何在Python嵌套列表中选取列？为何data[:][0]选中第一行？

首先明确你用data[:][0]无效的原因：data[:]只是创建了原列表的浅拷贝，和原data内容完全一致，后续的[0]还是取这个拷贝的第一个子列表（也就是第一行），自然达不到取列的效果。

下面是几种正确取列的方法：

方法1：列表推导式（基础Python实现，无需额外库） #

遍历每一行，提取对应位置的元素，比如取第一列：

first_col = [row[0] for row in data]
# 输出：[' 2.131e+02', ' 6.205e+01', '-4.220e-01']

要取第二列就把row[0]改成row[1]，以此类推。

方法2：用zip()函数转置列表 #

zip(*data)可以将原列表的行列转置，转置后每个元素对应原列表的一列，再转成列表操作：

# 转置得到所有列的元组
transposed = list(zip(*data))
# 取第一列并转成列表
first_col = list(transposed[0])
# 输出：[' 2.131e+02', ' 6.205e+01', '-4.220e-01']

如果需要一次性获取所有列，用[list(col) for col in zip(*data)]就能得到包含所有列的列表。

方法3：使用numpy（适合大规模数据操作） #

如果数据量较大或需要频繁做行列操作，推荐用numpy库：

import numpy as np

data_np = np.array(data)
# 用[:, 列索引]取所有行的对应列，最后转成列表
first_col = data_np[:, 0].tolist()
# 输出：[' 2.131e+02', ' 6.205e+01', '-4.220e-01']

numpy数组支持直接用[:, 列索引]的方式取列，直观又高效。

内容的提问来源于stack exchange，提问作者len