pandas dataframe与字典转换
import pandas as pd
df = pd.DataFrame({'Country': ['China','India', 'America', 'Japan'],
'Income': [10000, 5002, 20000, 5002],
'A': [50, 21, 34, 40]})
print(df)
Country Income A
0 China 10000 50
1 India 5002 21
2 America 20000 34
3 Japan 5002 40
df1 = df.set_index(['Country'])['Income'].to_dict()
print(df1)
{'China': 10000, 'India': 5002, 'America': 20000, 'Japan': 5002}
df2 = df.to_dict(orient = 'list')
print(df2)
{'Country': ['China', 'India', 'America', 'Japan'], 'Income': [10000, 5002, 20000, 5002], 'A': [50, 21, 34, 40]}