Pandasのデータフレームのヘッダ名を付けたい

Question

Excelファイルからpd.DataFrameとして読み込んだデータを重回帰分析したいのですが、こちらでヘッダ名を指定したく考えております。１列目に'y'、２列目以降のヘッダ名をx2,x3,...,x14に変更したく、
以下の様に行っております。

import pandas as pd
import numpy as np
import scipy as sp
import scipy.stats as stats
import statsmodels.api as sm
import statsmodels.formula.api as smf

cs=['y']
cs2=['x{}'.format(i+1) for i in range(14)] # x1...xnの列名を生成する
cs.append(cs2)
print(cs)
d.columns = cs

すると以下の様になってしまいます。

['y', ['x1', 'x2', 'x3', 'x4', 'x5', 'x6', 'x7', 'x8', 'x9', 'x10', 'x11', 'x12', 'x13', 'x14']]

ValueError Traceback (most recent call last)
<ipython-input-54-d97cbd168bcf> in <module>()
15 cs.append(cs2)
16 print(cs)
---> 17 d.columns = cs
18 #print(d.head())
19

~\Miniconda3\envs\stats\lib\site-packages\pandas\core\generic.py in setattr(self, name, value)
4383 try:
4384 object.getattribute(self, name)
-> 4385 return object.setattr(self, name, value)
4386 except AttributeError:
4387 pass

pandas/_libs/properties.pyx in pandas._libs.properties.AxisProperty.set()

~\Miniconda3\envs\stats\lib\site-packages\pandas\core\generic.py in _set_axis(self, axis, labels)
643
644 def _set_axis(self, axis, labels):
--> 645 self._data.set_axis(axis, labels)
646 self._clear_item_cache()
647

~\Miniconda3\envs\stats\lib\site-packages\pandas\core\internals.py in set_axis(self, axis, new_labels)
3321 raise ValueError(
3322 'Length mismatch: Expected axis has {old} elements, new '
-> 3323 'values have {new} elements'.format(old=old_len, new=new_len))
3324
3325 self.axes[axis] = new_labels

ValueError: Length mismatch: Expected axis has 15 elements, new values have 2 elements

対応につき、ご教示頂けますと幸いです。

Answer 1

リストの中にリストを追加しています。
cs.append(cs2)→cs = cs + cs2としてください。

Comments

[_hh]

できました！有難うございます。