2017-09-04 23 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]] 
>>>