如何移除pandas DataFrame列名中的公共重复词汇

你可以通过pandas内置的列名字符串处理方法快速实现需求，以下是两种常用实现方案：

方案1：固定字符串替换（最直观，适合已知要移除的固定内容的场景） #

直接对列名中的Student.和.word做字面量替换即可，代码如下：

import pandas as pd

# 原有数据构造逻辑
data = [['Mallika', 23, 'Student'], ['Yash', 25, 'Tutor'], ['Abc', 14, 'Clerk']]
data_frame = pd.DataFrame(data, columns=['Student.first.name.word', 'Student.Current.Age.word', 'Student.Current.Profession.word'])

# 核心修改列名逻辑
data_frame.columns = data_frame.columns.str.replace('Student.', '', regex=False).str.replace('.word', '', regex=False)

方案2：分割取中间段（更通用，适合前后缀固定的列名格式） #

你的列名都是前缀.需要保留的内容.后缀的结构，按.分割后取中间部分拼接即可，代码如下：

data_frame.columns = data_frame.columns.str.split('.').apply(lambda x: '.'.join(x[1:-1]))

两种方案运行后都可以得到你需要的列名格式的DataFrame。

内容的提问来源于stack exchange，提问作者British Bioinformatician