[英]Is there a more efficient way to write a dataframe (columns and data) to list?
我还没有找到将带有列和数据的数据框写入列表的功能。 我觉得必须要有一个函数,但是我找不到它,所以我将其串联在一起。
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]]
也许使用transpose
运算符可以解决问题?
df.T.reset_index().values.T.tolist()
[['Project ID', 'State', 'Cost'],
['Project1', 'CT', 800],
['Project2', 'MA', 1000],
['Project3', 'CA', 20]]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.