pandas删除最后一列_从Dataframe-Pandas中所有列的列名中删除最后两个字符

你可以这样用^{}In [214]: import functools as ft

In [215]: f = ft.partial(np.random.choice, *[5, 3])

In [225]: df = pd.DataFrame({'a': f(), 'b': f(), 'c': f(), 'a_1': f(), 'b_1': f(), 'c_1': f()})

In [226]: df

Out[226]:

a b c a_1 b_1 c_1

0 4 2 0 2 3 2

1 0 0 3 2 1 1

2 4 0 4 4 4 3

In [227]: df.columns = df.columns.str.rstrip('_1')

In [228]: df

Out[228]:

a b c a b c

0 4 2 0 2 3 2

1 0 0 3 2 1 1

2 4 0 4 4 4 3

但是，如果您需要更灵活的方法(尽管可能会慢一点)，可以使用^{}，借助regex的强大功能，它将允许您选择要保留列名的哪一部分In [216]: df = pd.DataFrame({f'{c}_{i}': f() for i in range(3) for c in 'abc'})

In [217]: df

Out[217]:

a_0 b_0 c_0 a_1 b_1 c_1 a_2 b_2 c_2

0 0 1 0 2 2 4 0 0 3

1 0 0 3 1 4 2 4 3 2

2 2 0 1 0 0 2 2 2 1

In [223]: df.columns = df.columns.str.extract(r'(.*)_\d+')[0]

In [224]: df

Out[224]:

0 a b c a b c a b c

0 1 1 0 0 0 2 1 1 2

1 1 0 1 0 1 2 0 4 1

2 1 3 1 3 4 2 0 1 1

使用df.columns.str的想法来自this答案