如何通过空格分割重命名 pandas 列名

Question

我有一个像这样的数据框：

index   col A   col B   col C
index1    3       1       2
index2    1       4       9
index3    5       1       2
index4    8       2       2
index5    2       1       6

我想通过用空格分割列来重命名它们。我不想手动执行此操作，因为我有数百列。我的输出如下所示：

index     A       B       C
index1    3       1       2
index2    1       4       9
index3    5       1       2
index4    8       2       2
index5    2       1       6

谁能帮我解决这个问题

Answer 1

用途：

df.columns = df.columns.str.split().str[1]

Answer 2

你可以

extract

最后一句话：

df.columns = df.columns.str.extract(r'(\S+)$', expand=False)

输出：

    index  A  B  C
0  index1  3  1  2
1  index2  1  4  9
2  index3  5  1  2
3  index4  8  2  2
4  index5  2  1  6