Remove column name suffix from DataFrame Python

Question

I have the following pandas DataFrame in python:

attr_x	header	example_x	other
3232	322	abv	ideo
342	123213	ffee	iie
232	873213	ffue	iie
333	4534	ffpo	iieu

I want to remove the suffixes '_x' from all columns containing it. The original DataFrame is much longer. Example result:

attr	header	example	other
3232	322	abv	ideo
342	123213	ffee	iie
232	873213	ffue	iie
333	4534	ffpo	iieu

jezrael · Accepted Answer · 2022-10-03 11:24:09Z

4

df.columns = df.columns.str.removesuffix("_x")

Or replace:

df.columns = df.columns.str.replace(r'_x$', '')

answered Oct 3, 2022 at 11:24

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

This for python 3.10 I believe