读取csv文件的列,并使用Python将每列保存在变量中

时间:2017-06-23 17:47:17

标签: python csv

我想将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相同的价值)。 你能帮帮我吗?

1 个答案:

答案 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']