python - 读取csv文件的列，并使用Python将它们分别保存在变量中

我想将csv文件的每一列保存到一个变量中（例如：在“ varA”变量中称为“ a”的列，在“ varB”变量中称为“ b”的列，等等）。
我试图用这种方式做到这一点：

with open(data_file_name, 'r') as csvfile:
    varA = [int(row["a"]) for row in DictReader(csvfile)]
    varB = [float(row["b"]) for row in DictReader(csvfile)]
    lenA = len(varA)
    lenB = len(varB)

我已经亲自检查了文件，并且所有列的长度都相同，但是在此代码中，我可以获得具有特定值的lenA，但没有lenB的值（因为当我打印lenB时，我看到的是0而不是相同的值lenA）。
你能帮我吗？

最佳答案

使用熊猫：

file.csv：

a,b,c
1,0,1
1,0,0
0,0,1
1,0,0

做：

import pandas as pd

df = pd.read_csv('file.csv')

var_a = df['a']
var_b = df['b']
var_c = df['c']

关于python - 读取csv文件的列，并使用Python将它们分别保存在变量中，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/44727249/