我有一個文件「data.csv」,看起來像Python的讀取csv文件列進名單,無視頭
ColA, ColB, ColC
1,2,3
4,5,6
7,8,9
我想打開和讀取文件列進名單,與第1項該列表被省略,例如
dataA = [1,4,7]
dataB = [2,5,8]
dataC = [3,6,9]
實際上有超過3列,列表很長,這只是格式的一個例子。我已經試過:
csv_file = open('data.csv','rb')
csv_array = []
for row in csv.reader(csv_file, delimiter=','):
csv_array.append(row)
在那裏我會那麼csv_array
各指標分配到一個列表,例如
dataA = [int(i) for i in csv_array[0]]
但我發現了錯誤:
_csv.Error: new-line character seen in unquoted field - do you need to open the file in universal-newline mode?
而且感覺很長篇大論只是將數據保存到幾個列表的方式...
謝謝!
編輯:
這是我如何解決它:
import pandas as pd
df = pd.read_csv('data.csv', names = ['ColA','ColB','ColC']
dataA = map(int,(df.ColA.tolist())[1:3])
,並重複列的其餘部分。
如果你不介意使用進口的,我會建議使用tablib,你可以在這裏找到:http://docs.python-tablib.org/en/master/ – rak007