我想将csv文件的每一列保存到一个变量中(例如:' a''变量'变量中的列,' b' ;在' varB'变量等)。 我试图这样做:
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相同的价值)。 你能帮帮我吗?
答案 0 :(得分:2)
使用pandas:
<强> 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']