假设我有一个 CSV 文件。 它包含:
a,b,c
1,2,3
4,5,6
7,8,9
如何将标有“a”的列转换为列表,而无需对其行进行硬编码?
最初,我为第一行 a,b,c 做一个 readline,然后为每个变量硬编码一行。但后来我想:如果我想这样做而 CSV 文件的排列方式可能不一样怎么办 - 说:
b,a,c
2,1,3
5,4,6
8,7,9
解决这个问题的最佳方法是什么?
请您参考如下方法:
您可以使用 pandas 执行此操作图书馆:
import pandas
data = pandas.read_csv("test.csv", header=0)
col_a = list(data.a)
col_b = list(data.b)
col_c = list(data.c)
col_a、col_b和col_c中的值如下:
[1, 4, 7]
[2, 5, 8]
[3, 6, 9]
