2017-09-04 35 views
0

我最初加載CSV文件的Python腳本:大熊貓據幀以「傳統」文件加載

data= [] 
with open('data.csv', 'r') as f: 
    for line in f.readlines()[1:]: 
     data.append(re.split(r',',line.rstrip())) 

但現在我動態在什麼樣的結果產生蟒之內這個CSV的內容是熊貓DataFrame。我的問題是什麼是將數據轉換爲所需格式(列表清單)的最有效方式。

一個顯而易見的方法是將數據幀保存到csv,然後像上面那樣加載它,另一個以某種方式直接轉換。該方法是否要走下去?

data = df.values.tolist() 

回答

1

如你所說,如果你有一個熊貓據幀df,您可以使用df.values.tolist()獲得值列表的列表。

下面是一個例子:

>>> import pandas as pd 
>>> df = pd.DataFrame([[1,2],[3,4],[5,6]]) 
>>> df.values 
array([[1, 2], 
     [3, 4], 
     [5, 6]]) 
>>> df.values.tolist() 
[[1, 2], [3, 4], [5, 6]] 
>>>