为所有数据框列名添加一个数字，但列名和由字符串和整数组成

Question

我需要通过在名称内的所有数字中添加1来重命名我的所有列名称

问题是，每个列名都由文本和数字的多个部分组成，我希望在输出时保留这些部分

下面的输入示例：

^{tb1}$

所需输出为：

^{tb2}$

为了节省时间，您可以使用以下命令生成示例输入：

data = [[1, 2, 4], [3, 5, 4], [2, 7, 6]] 
df = pd.DataFrame(data, columns = ["ARG15 - ILE10 vdW", "VAL16 - ILE10 vdW", "VAL16 - VAL19 vdW"]) 

如果有帮助的话，我在现实生活中的列名都具有与上面所示相同的格式：“text”“number”“-”“text”“number”“text”，因此一个更具体的解决方案就可以了

（我正在使用python 3.8）

谢谢

Answer 1

您已经注意到，列是固定格式的

• 转置，因为操作行比操作列更简单
• 使用正则表达式解析出组件
• 根据您的需求重新构建
• set_index()具有新值
• 转置以返回到原始结构

import re
myre = re.compile("^([A-Z]+)([0-9]+) - ([A-Z]+)([0-9]+) ([A-Z,a-z]+)$")

data = [[1, 2, 4], [3, 5, 4], [2, 7, 6]] 
df = pd.DataFrame(data, columns = ["ARG15 - ILE10 vdW", "VAL16 - ILE10 vdW", "VAL16 - VAL19 vdW"]) 
df = (df.T.assign(bits=lambda dfa: dfa.index)
 .assign(bits=lambda dfa: dfa.bits.apply(lambda s: "".join([f"{a}{int(b)+1} - {c}{int(d)+1} {e}" 
                                                            for a, b, c, d, e in re.findall(myre, s)])))
 .set_index("bits")
 .T
)

^{tb1}$