python - 根据字典重命名 Pandas 中的列

Question

我有一个 pandas DataFrame，我想根据我计划用作字典的另一个 DataFrame 重命名列。

例如第一个DataFrame是:

          AAA   BBB   CCC   DDD
 index   
  1       1     2     3     4
  2       5     6     7     8

和我想用作字典的第二个 DataFrame:

           val1    val2
  index
    1      AAA      A7
    2      BBB      B0
    3      CCC      C3
    4      DDD      D1

我想得到的结果如下:

          A7    B0    C3    D1 
 index   
  1       1     2     3     4
  2       5     6     7     8

最初我想将第一个 DataFrame reshape 为长格式，然后与字典 DataFrame 合并，然后 reshape 回宽格式。但是我认为这是非常低效的，所以我想使用一种更有效的方法(如果存在的话)。非常感谢四位的帮助。

Answer 1

df.rename有一个名为 columns 的参数，它接受字典:

df.rename(columns=dict(zip(df2["val1"], df2["val2"])))

输出:

    A7  B0  C3  D1
0   1   2   3   4
1   5   6   7   8

它返回一个新的 DataFrame。您可以使用 inplace=True，或将其分配回原始 DataFrame。