Pandas 交换列名中的子字符串

Question

的列名称中的子字符串

我尝试交换 DataFrame K_1[0,0]	K_1[0,1]	K_1[1,20]
12	34	77
99	42	23

，结果应如下所示

K_1[0,0]	K_1 [1,0]	K_1[20,1]
12	34	77
99	42	23

所以括号内的值逗号前后应该简单地互换。

提前致谢！

Answer 1

您可以将 DataFrame.rename 与使用正则表达式的可调用函数一起使用。

import re
df = df.rename(lambda c: re.sub('(.*)(\d+),(\d+)(.*)', r'\1\3,\2\4', c), axis=1)

Answer 2

使用 str.replace< /代码>：

df = pd.DataFrame(columns=['K_1[0,0]','K_1[0,1]','K_1[1,20]'])

df.columns = df.columns.str.replace(r'^(.*)(\d+),(\d+)(.*)
, r'\1\3,\2\4')
print (df)      
Empty DataFrame
Columns: [K_1[0,0], K_1[1,0], K_1[20,1]]
Index: []