带有自定义列名的Python pandas.DataFrame.to_dict

Posted 2021-04-01

当我在我的数据帧上运行df.to_dict（'records'）时，我得到了这个：

    {'country_name': 'Australia', 'count': 55},
    {'country_name': 'Brunei Darussalam', 'count': 90},
    {'country_name': 'Cambodia', 'count': 234}

但是，这就是我想要的：

    {'name': 'Australia', 'y': 55},
    {'name': 'Brunei Darussalam', 'y': 90},
    {'name': 'Cambodia', 'y': 234}

是否有捷径可寻？

答案

在你做to_dict之前，rename吧

df.rename(columns={'country_name':'name','count':'y'}).to_dict('records')