Python中将dataframe转换为字典

有时候,在Python中需要将dataframe类型转换为字典类型,下面的方法帮助我们解决这一问题。
任务代码

# encoding: utf-8

import pandas as pd
a = ['Name', 'Age', 'Gender']
b = ['Ali', '19', 'China']
data = pd.DataFrame(zip(a, b), columns=['project', 'attribute'])
print data
dict_country = data.set_index('project').T.to_dict('list')
print dict_country

输出显示

  project attribute
0    Name       Ali
1     Age        19
2  Gender     China
{'Gender': ['China'], 'Age': ['19'], 'Name': ['Ali']}

值得注意的是,转置之前需要设置指定的索引,否则会按照默认索引转换成这样:

{0: ['Name', 'Ali'], 1: ['Age', '19'], 2: ['Gender', 'China']}

作者:戴翔
电子邮箱:daixiangcn@outlook.com


daixiangcn CSDN认证博客专家 后端开发工程师 产品体验官
中国公民,中共预备党员,CSDN博客专家,本科大四在读。江苏省第四届“互联网+”大学生创新创业大赛二等奖获得者,第二届、第三届微信小程序应用开发大赛全国三等奖、华东赛区二等奖项目作者,微信小程序《约车吗》、《庆云慈善会》、《锋云速检》项目工程师。
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 技术工厂 设计师:CSDN官方博客 返回首页