pandasデータフレームの一部の列名の変更

Question

example_df = pd.DataFrame([
                ['John', 20, 45, 30, 50], 
                ['Peter', 21, 62, 40, 100], 
                ['Scot', 25, 68, 100,200]],

                columns=['Name', 'Age', 'Marks','Height','Weight'])

example_df

Name Age Marks height Weight
0 John 20 45 30 50
1 Peter 21 62 40 100
2 Scot 25 68 100 200

このMarks, height, Weightをfactor1, factor2, factor3...(実際に無数のカラムがある)にしたいです。

下のようにしても結果が変わりませんでした。

example_df.iloc[:,2:].columns = example_df.iloc[:,2:].columns.map(lambda x:'factor') #ここのナンバリングも課題
example_df
>> Name 	Age 	Marks 	height 	Weight
0 	John 	20 	45 	30 	50
1 	Peter 	21 	62 	40 	100
2 	Scot 	25 	68 	100 	200

どうしたらいいでしょうか。

Answer 1

案1. rename用の辞書を作ってrenameする。

col_mapper = {col: f'factor{i}'
              for i, col in enumerate(example_df.columns[2:], 1)}

example_df = example_df.rename(columns=col_mapper)

案2. カラム名をlistに変換して、そのlistを変更してから、カラム名にセットしなおす。

col_list = example_df.columns.to_list()
col_list[2:] = [f'factor{i}' for i in range(1, len(col_list[2:]) + 1)]
example_df.columns = col_list

あとで、factorXXの元のカラム名が何だったかとかを見る必要があるなら、案1のほうが良いかもしれません。

Comments

[sasakin]

辞書内でも内包表記？ができるのですね！
スマートすぎる方法に感動しました！！
本当に感謝です、ありがとうございました。

Answer 2

以下をご覧ください。

pandas.DataFrameの行名・列名の変更

Comments

[sasakin]

すいませんが、こちら既に確認しましたが、
辞書で指定するのではなく、dfの特定の列(例えば3-5)に対してprefix fatorをもつ数字の並び順(factor 1,fator2, ..)で作成したいです。