1. ホーム
  2. pandas

[解決済み] Pandas: 複数インデックスを持つカラムのデータフレームにカラムを追加する

2023-04-01 16:26:36

質問

マルチインデックス列データフレームの第2階層に列を追加したい。

In [151]: df
Out[151]: 
first        bar                 baz           
second       one       two       one       two 
A       0.487880 -0.487661 -1.030176  0.100813 
B       0.267913  1.918923  0.132791  0.178503
C       1.550526 -0.312235 -1.177689 -0.081596 

通常の直接代入のトリックではうまくいきません。

In [152]: df['bar']['three'] = [0, 1, 2]

In [153]: df
Out[153]: 
first        bar                 baz           
second       one       two       one       two 
A       0.487880 -0.487661 -1.030176  0.100813
B       0.267913  1.918923  0.132791  0.178503
C       1.550526 -0.312235 -1.177689 -0.081596

3行目を"bar"の下に追加するにはどうしたらよいですか?

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

実はとても簡単です(参考までに、私はもともとあなたの方法でやろうと思っていました)。

df['bar', 'three'] = [0, 1, 2]
df = df.sort_index(axis=1)
print(df)

        bar                        baz          
        one       two  three       one       two
A -0.212901  0.503615      0 -1.660945  0.446778
B -0.803926 -0.417570      1 -0.336827  0.989343
C  3.400885 -0.214245      2  0.895745  1.011671