是否有Pandas函数可以忽略输入Excel文件中同名的列

eqoofvh9  于 2023-06-25  发布在  其他
关注(0)|答案(1)|浏览(95)

我有一个excel文件,其中包含一些同名列,并希望在熔化数据之前将其导入到多个数据框架中。我如何在数据框架中拥有原始列的名称?问题是,我的数据框架的列名为ex:A,A.1,A.2。现在我重命名的列,但没有一个更好的方法。
我试过重命名列,但它太长了(我有超过30列的重复和重命名超过10次),我想一个更快的方法或函数做得更好

kcrjzv8t

kcrjzv8t1#

没有“开箱即用”的方法可以做到这一点,但是你可以用str.replace删除这个后缀:

df.columns = df.columns.str.replace(r'\.\d+$', '', regex=True)

示例(使用read_csv):

import pandas as pd
import io

df = pd.read_csv(io.StringIO('A,B,C,A,B,C\n1,2,3,4,5,6'))
df.columns = df.columns.str.replace(r'\.\d+$', '', regex=True)
print(df)

输出:

A  B  C  A  B  C
0  1  2  3  4  5  6

相关问题