1. ホーム
  2. python

[解決済み] リセットインデックスに名前を付けることはできますか?

2023-02-03 06:53:24

質問

通常、データフレームが reset_index() が行われると、新しいカラムには index または level_i を使用します。

新しいカラムに名前を付けることは可能ですか?

どのように解決するのですか?

以下のように rename から返された df に対して reset_index :

In [145]:
# create a df
df = pd.DataFrame(np.random.randn(5,3))
df

Out[145]:
          0         1         2
0 -2.845811 -0.182439 -0.526785
1 -0.112547  0.661461  0.558452
2  0.587060 -1.232262 -0.997973
3 -1.009378 -0.062442  0.125875
4 -1.129376  3.282447 -0.403731

インデックス名を設定する

In [146]:    
df.index = df.index.set_names(['foo'])
df

Out[146]:
            0         1         2
foo                              
0   -2.845811 -0.182439 -0.526785
1   -0.112547  0.661461  0.558452
2    0.587060 -1.232262 -0.997973
3   -1.009378 -0.062442  0.125875
4   -1.129376  3.282447 -0.403731

コール reset_index と連鎖し rename :

In [147]:
df.reset_index().rename(columns={df.index.name:'bar'})

Out[147]:
   bar         0         1         2
0    0 -2.845811 -0.182439 -0.526785
1    1 -0.112547  0.661461  0.558452
2    2  0.587060 -1.232262 -0.997973
3    3 -1.009378 -0.062442  0.125875
4    4 -1.129376  3.282447 -0.403731

ayhanに感謝

別の方法として rename_axis の前にインデックスの名前を変更するために reset_index :

In [149]:
df.rename_axis('bar').reset_index()

Out[149]:
   bar         0         1         2
0    0 -2.845811 -0.182439 -0.526785
1    1 -0.112547  0.661461  0.558452
2    2  0.587060 -1.232262 -0.997973
3    3 -1.009378 -0.062442  0.125875
4    4 -1.129376  3.282447 -0.403731

とか、最初にインデックス名を直接上書きしてしまえばいいのです。

df.index.name = 'bar'

を呼び出して reset_index