python - 在 Pandas DataFrame 中使列和排序保持一致

Question

我正在寻找使 Pandas DataFrame 列保持一致的优雅的 Pythonic 方式。含义:

1. 确保主列表中的所有列都存在，如果不存在，则添加一个空的占位符列。
2. 确保列的顺序与主列表的顺序相同。

我有以下有效的示例，但是是否有用于实现相同目标的内置 Pandas 方法？

import pandas as pd
df1 = pd.DataFrame(data=[{'a':1,'b':32, 'c':32}])
print df1

   a   b   c0  1  32  32

column_master_list = ['b', 'c', 'e', 'd', 'a']
def get_dataframe_with_consistent_header(df, headers):
    for col in headers:
        if col not in df.columns:
            df[col] = pd.np.NaN
    return df[headers]

print get_dataframe_with_consistent_header(df1, column_master_list)

   b   c   e   d   a0 32  32 NaN NaN   1

Answer 1

您可以使用 reindex方法。传入列名列表并指定 'columns'。默认情况下，缺失条目的填充值为 NaN:

>>> df1.reindex(column_master_list, axis='columns')
    b   c   e   d  a
0  32  32 NaN NaN  1