Разделить имя столбца Pandas по разделителю

Question

У меня есть фрейм данных с заголовками столбцов, которые я хочу разделить разделителем '__' и использовать только первую часть в качестве имени столбца.

Например,

start

    Red__34343    Green__485838384    Blue__3
0
1
2
3

вывод

    Red    Green    Blue
0
1
2
3

Answer 1

Вы можете попробовать extract:

df.columns = df.columns.str.extract('(.*)__')[0]

вывод:

0  Red  Green  Blue
0    0    NaN   NaN
1    1    NaN   NaN
2    2    NaN   NaN
3    3    NaN   NaN

Answer 2

Использовать str.split:

df.columns=df.columns.str.split('__',expand=True).get_level_values(0)
df.columns
#Index(['Red', 'Green', 'Blue'], dtype='object')

или как @ anky_91 предлагается

df.columns=df.columns.str.split('__').str[0]