2017-06-23 8 views
0

csvファイルの各列を変数に保存します(例: 'varA'変数の 'a'という列、 'varB'変数の 'b'など)。 私はこのように、それを行うことを試みた:csvファイルの列を読み込み、Pythonを使用して各列を変数に保存します。

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を表示したときにlenAと同じ値の代わりに0が表示されるためです)。 私を助けてもらえますか?

答えて

1

利用パンダ:

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'] 
関連する問題