データフレームの文字列を置換したい

Question

前提・実現したいこと

Python初心者です。ただいま正規表現を使い、データフレームの文字列を置換するコードを書こうとしています。具体的には、データフレームのカラムに特定の文字列が含まれる場合、それを削除したいと考えています。

発生している問題・エラーメッセージ

データフレームの内容を上書きできず、元のカラム名のままになってしまいます。

該当のソースコード

# dfというデータフレームのカラム名が「_x」を含む場合、それを削除
for d in df.columns:
  if '_x' in d:
    d = re.sub('_x', '', d)

print(df.columns)

試したこと

for文で回している時点で何か間違えているのではないかと思い、print()で中身を出力しましたが、この時点では正確に置換ができていました。正規表現ではなくreplace()を用いて同じことをやってみましたが、これも同じ結果に終わりました。問題は、上書きのやり方がわかっていない点にあるのだと考えています。どのようにすればよいでしょうか。初歩的な質問で恐縮ですが、ご教示いただけますと幸いです。

補足情報（FW/ツールのバージョンなど）

GoogleColabを用いて作業をしています。

Answer 1

indexの要素である文字列から'_x'を削除した文字列を作成しても、それは文字列を変更したことにはなりません。文字列は変更不能(イミュータブル)だからです。
以下のようにインデックス全体を置き換える必要があります。

import re
df.columns = [re.sub('_x', '', c) for c in df.columns]

実行結果

>>> print(df)
   i_x  _x_xj   _xk
0    0      0  90.0
1    2      0  92.0
2    1      1  91.1
3    1      0  91.0
4    2      2  92.2
>>> df.columns = [re.sub('_x', '', c) for c in df.columns]
>>> print(df)
   i  j     k
0  0  0  90.0
1  2  0  92.0
2  1  1  91.1
3  1  0  91.0
4  2  2  92.2

１個づつやりたければ、pandas.DataFrameの行名・列名の変更に書かれているrenameを使えば不可能ではありませんが、上記の方法よりもかなり複雑になります。

Comments

[k_matsumot]

ご回答、感謝いたします。自分がどこを理解していなかったか、よくわかりました。教えていただいたとおり、内包表記で書いてインデックス全体に代入したところ、うまく動きました。丁寧に教えてくださり、ありがとうございました。