根据查找替换熊猫数据框中的列名

时间:2018-07-17 13:28:41

标签: pandas dataframe

嗨,我有几个列标题略有不同的数据框。数据框标头的示例为:

A1    B1    C1

在其他数据框中,第一行称为A2或A3等。A1...,B1 ... C1表示多字符单词/标签,而不是文字列名称。我想根据我在A1,A2,Ax,B1,B2,Bx,C1,C2,Cx等与A,B和C之间的映射替换列标题。

做到这一点的最佳方法是什么?

谢谢。

1 个答案:

答案 0 :(得分:1)

我认为这里可以用indexing with str替换首字母:

df.columns = df.columns.str[0]

另一种可能的解决方案是创建替换字典,例如:

d = {x:x[0] for x in df.columns}
df = df.rename(columns=d)