Pandasで多数のヘッダ名を付与したい

Question

Jupyter notebookでPythonを利用しております。あるcsvファイルを読み込み、重回帰分析を行いたいのですが、独立変数には本来のヘッダの名称ではなく、連番の名前を付けたく、データの列数が少ない時には、例えば以下の様にしております。

d = pd.read_csv('slump_test.data.txt', sep=',',header=None,skiprows=1,names=('x1','x2','x3','x4','x5','x6','x7','x8','x9','x10','y'))

しかし、列数が数百あるような場合にも、同様にしたいのですが、その場合どのようにしたらよいでしょうか。例えば301列ある場合には、x1からx300までと、yという変数名をヘッダにしたいのですが。
方法ございましたら、ご教示頂けますと幸いです。

Answer 1

以下のような感じで列名を変更できます。

import pandas as pd
d = pd.DataFrame({'a':[1,2],'b':[3,4],'c':[5,6]}) # 読み込んだデータ
print(d)
print(d.columns) # Index(['a', 'b', 'c'], dtype='object')

cs = ['x{}'.format(i+1) for i in range(len(d.columns)-1)] # x1...xnの列名を生成する
cs.append('y') # 末尾は「y」
d.columns = cs
print(d)
"""
   x1  x2  y
0   1   3  5
1   2   4  6
"""
print(d.columns) # Index(['x1', 'x2', 'y'], dtype='object')

Comments

[_hh]

できました！ありがとうございます。