Jac*_*ler 2 python list dataframe pandas
我需要将 DataFrame 转换为列表,并将其标题保存为列表中的第一项。我想其中一种方法是分别转换列和值,然后加入这些列表。但还有更好的办法吗?
数据框,
import pandas as pd
df=pd.DataFrame({"Sales QTY":[10,20,30,40],
"Sales Person":['Jack', 'Adam', 'Ken', 'Jack'],
"Product":["Apple", "Orange","Apple","Cherry"]
})
Run Code Online (Sandbox Code Playgroud)
期望的输出
[['Sales QTY', 'Sales Person', 'Product'],
[10, 'Jack', 'Apple'],
[20, 'Adam', 'Orange'],
[30, 'Ken', 'Apple'],
[40, 'Jack', 'Cherry']]
Run Code Online (Sandbox Code Playgroud)
lst = df.to_numpy().tolist()
lst.insert(0, df.columns.tolist())
lst
[['Sales QTY', 'Sales Person', 'Product'],
[10, 'Jack', 'Apple'],
[20, 'Adam', 'Orange'],
[30, 'Ken', 'Apple'],
[40, 'Jack', 'Cherry']]
Run Code Online (Sandbox Code Playgroud)