python - 如何将列名从元组更改为字符串？

Question

我在 pandas dataframe 中使用了 pd.pivot_table，列名变成了 ('A1', 'B1'), ('A1', 'B2') 这样的元组。 .. 我希望它们像 'A1_B1', 'A1_B2'...

我试着用df.columns.values[i] = df.columns.values[i][0] + '_' + df6.columns.values[i][1],并尝试重命名。

当我检查 df.columns.values 时，列的名称发生了变化，但是当我无法使用这些名称进行索引时。我是 python 新手，所以可能不知道列名和列索引之间的区别。

谁能帮帮我？谢谢!

Answer 1

设置

df = pd.DataFrame(
    np.arange(8).reshape(2, 4),
    columns=[('A1', 'B1'), ('A2', 'B1'), ('A1', 'B2'), ('A2', 'B2')])

print(df)

   (A1, B1)  (A2, B1)  (A1, B2)  (A2, B2)
0         0         1         2         3
1         4         5         6         7

重命名

df.rename(columns='_'.join, inplace=True)
print(df)

   A1_B1  A2_B1  A1_B2  A2_B2
0      0      1      2      3
1      4      5      6      7

map

df.columns = df.columns.map('_'.join)
print(df)

   A1_B1  A2_B1  A1_B2  A2_B2
0      0      1      2      3
1      4      5      6      7