一日一技:Pandas 如何對列排序?
作者: kingname
我們在工作中,經常用到 Excel,有時候,我們會使用 Pandas 生成 Excel。但生成的 Excel 列的順序可能跟我們想要的不一樣。
我們在工作中,經常用到 Excel,有時候,我們會使用 Pandas 生成 Excel。但生成的 Excel 列的順序可能跟我們想要的不一樣。
例如:
- import pandas as pd
- datas = [
- {'id': 1, 'name': '王大', 'salary': 9999, 'work_time': 19},
- {'id': 2, 'name': '李二', 'salary': 9999, 'work_time': 19},
- {'id': 3, 'name': '張三', 'salary': 9999, 'work_time': 19},
- {'id': 4, 'name': '朱四', 'salary': 9999, 'work_time': 19},
- {'id': 5, 'name': '陳五', 'salary': 9999, 'work_time': 19},
- {'id': 6, 'name': '老牛', 'salary': 9999, 'work_time': 19},
- ]
- df = pd.DataFrame(datas)
- df.to_excel('example.xlsx', index=False)
運行效果如下圖所示:
現在,我想在最終生成的 Excel 中,把work_time放到salary左邊。這個時候,有兩種方案:
方法1,把包含正確列表順序的列表,傳給 DataFrame 對象。
- df = df[['id', 'name', 'work_time', 'salary']]
運行效果如下圖所示:
方法2,使用.reindex()方法:
- df = df.reindex(columns=['id', 'name', 'work_time', 'salary'])
運行效果如下圖所示:
責任編輯:武曉燕
來源:
未聞Code