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

Question

当我在数据帧上运行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}

有一种简单的方法吗？

Answer 1

在进行to_dict之前，rename

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