是否有更有效的方法将数据帧(列和数据)写入列表?

时间:2015-07-29 17:45:40

标签: python python-2.7 pandas

我还没有找到一个函数来将包含列和数据的数据帧写入列表。我觉得必须有一个功能,但我找不到它,而是我连在一起。

import pandas as pd
df = pd.DataFrame(data = [['Project1', 'CT', 800], ['Project2', 'MA', 1000], ['Project3', 'CA', 20]], columns=['Project ID', 'State', 'Cost'])

print [df.columns.tolist()] + df.values.tolist()

期望的输出:

[['Project ID', 'State', 'Cost'], ['Project1', 'CT', 800L], ['Project2', 'MA', 1000L], ['Project3', 'CA', 20L]]

1 个答案:

答案 0 :(得分:1)

也许使用transpose运算符来做这个伎俩?

df.T.reset_index().values.T.tolist()

[['Project ID', 'State', 'Cost'],
 ['Project1', 'CT', 800],
 ['Project2', 'MA', 1000],
 ['Project3', 'CA', 20]]