python - 从 Dataframe 中所有列的列名中删除最后两个字符 - Pandas

Question

我正在使用用户 ID 键加入具有相同列/列名称的两个数据框 (a,b)，并且在加入时，我必须提供后缀字符，以便创建它。以下是我使用的命令，

a.join(b,how='inner', on='userId',lsuffix="_1")

如果我不使用这个后缀，我就会出错。但我不想更改列名，因为这会在运行其他分析时导致问题。所以我想从结果数据框的所有列名中删除这个“_1”字符。谁能建议我一种有效的方法来删除 Pandas 数据框中所有列名称的最后两个字符？

谢谢

Answer 1

这段代码应该可以完成工作:

df.columns = pd.Index(map(lambda x : str(x)[:-2], df.columns))

编辑:这是一种更好的方法

df.rename(columns = lambda x : str(x)[:-2])

在这两种情况下，我们所做的只是遍历列并应用一些函数。在这种情况下，该函数将某些内容转换为字符串，并获取所有内容，直到最后两个字符。

我敢肯定还有其他几种方法可以做到这一点。