Pandas列重命名：移除分隔符~前文本，解决索引越界问题

问题场景 #

你的DataFrame列名如下：

['user_id', 'Week 36~Sep-05 - Sep-11', 'Week 35~Aug-29 - Sep-04', 'Week 34~Aug-22 - Aug-28']

想要移除~之前的文本得到目标列名，但使用[col.split('~')[1] for col in df.columns]时触发IndexError: list index out of range。

报错原因 #

user_id这个列名里没有~，执行split('~')后只会得到长度为1的列表，此时尝试取索引[1]自然会越界。

解决方案 #

提供三种可行的处理方式：

1. 列表推导式加条件判断

直接在推导式里判断列名是否包含~，按需处理：

df.columns = [col.split('~')[1] if '~' in col else col for col in df.columns]

2. Pandas字符串方法（推荐）

利用Pandas的字符串处理API，更简洁且适配所有列：

df.columns = df.columns.str.split('~').str.get(-1)

这里str.get(-1)会取每个拆分后列表的最后一个元素——带~的列取到的是~后的内容，无~的列直接保留原列名。

3. 正则表达式替换

用正则匹配并删除~及之前的所有内容：

df.columns = df.columns.str.replace(r'^.*~', '', regex=True)

正则^.*~会匹配从列名开头到第一个~的所有字符，替换为空后就得到目标内容，无~的列不受影响。

内容的提问来源于stack exchange，提问作者Kevin Nash