我想将 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)。你能帮我吗?
使用熊猫:
文件.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']
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句