Python二维列表取一列的多方面阐述

发布时间：2023-05-21

一、基本语法

list_name[column_num]

通过在二维列表的名称后添加方括号，再在方括号内添加欲取出的列数，即可实现对某一列数据的提取。例如：

data = [['apple', 'banana', 'cherry'],[1, 2, 3],[4, 5, 6]]
column_1 = [row[0] for row in data]  #提取第一列
print(column_1)

上面的代码会输出:

['apple', 1, 4]

其中，列表推导式[row[0] for row in data]是一种简洁的写法，它等效于下面的代码：

column_1 = []
for row in data:
    column_1.append(row[0])

在实际应用中，我们可以通过循环遍历整个二维列表，然后取出每一行指定位置的数据。

二、获取多列数据

如果需要同时获取多列数据，我们可以将上述方法进行适当修改：

column_1, column_2 = [], []
for row in data:
    column_1.append(row[0])
    column_2.append(row[1])
print(column_1)
print(column_2)

上述示例可以同时获取第一列和第二列的数据。如果需要获取更多列的数据，只需要再添加一个空列表，并在循环中为其添加数据即可。

三、获取特定列数据

有时候我们需要获取二维列表中特定列的数据，可以使用list comprehension即列表推导式配合zip()函数来实现：

selected_cols = [1, 2]  # 待选择的列
result = [list(col) for col in zip(*[data[i] for i in selected_cols])]
print(result)

上述代码中，zip()函数用于将对应列的数据进行组合, selected_cols则指定了我们需要获取哪些列的数据。 运行上述代码后，我们可以获得二维列表中第二列和第三列的数据:

[[1, 4], [2, 5], [3, 6]]

四、切片操作

如果我们只需要二维列表中的一部分数据，我们可以使用切片操作来完成：

result = [row[1:3] for row in data]
print(result)

上述代码中，row[1:3]表示对于每一行，我们只需要第二列和第三列的数据。 运行上述代码后，我们可以获得:

[[1, 2], [4, 5]]

五、Numpy库的使用

使用Python提供的Numpy库，可以更加方便地进行二维列表的处理。比如，我们可以使用numpy.transpose()函数进行转置操作，将二维列表的行列互换:

import numpy as np
data = np.array(data)
data = data.transpose()
print(data)

上述代码中，我们先使用numpy.array()函数将Python列表转化为Numpy数组，然后通过transpose()函数进行转置操作。 运行上述代码后，我们可以获得:

[['apple' 1 4]
 ['banana' 2 5]
 ['cherry' 3 6]]

六、Pandas库的使用

使用Pandas库可以更加高效、便捷地进行二维列表的处理和分析。我们可以通过pandas.DataFrame()函数将二维列表转化为数据框:

import pandas as pd
df = pd.DataFrame(data)
print(df)

上述代码中，我们使用pandas.DataFrame()函数将二维列表转化为数据框，然后对数据框进行分析和处理。 运行上述代码后，我们可以获得:

      0  1  2
0 apple  1  4
1 banana 2  5
2 cherry 3  6

除了使用Pandas库之外，我们还可以使用其他数据分析和处理库来对二维列表进行更加深入的分析和处理，例如Numpy、Matplotlib等等。

推荐文章