python・pandasで縦持から横持に変換後のカラム名を変更するにはどうすればよいでしょうか

Question

Python　pandasのデータフレームの縦持から横持への変換時、カラム名を設定する方法はどうしたらいいのでしょうか？
以下のデータをデータフレームdfとして持っていたとします。
id year bw
0 1 1 50
1 1 2 45
2 1 3 55
3 2 1 34
4 2 2 33
5 3 1 67
6 3 2 66
7 3 3 69
8 3 4 72

df2=df.pivot(values=['bw'], columns=['year'], index=['id'])
で変換すると、

bw

year 1 2 3 4
id
1 50.0 45.0 55.0 NaN
2 34.0 33.0 NaN NaN
3 67.0 66.0 69.0 72.0

となります。このyear 1 2 3 4となっているカラム名はどのように扱えばよいのでしょうか？
year=1の列、
50.0
34.0
67.0
を取り出すのに、df2['1']としても取り出せないし、カラム名をどのように設定すればよいのかわかりません。
また他のカラム名に変更しようとしても、元の名前がわからないと、変更できませんし・・。

どなたかご教示お願いします

Answer 1

たとえばdf2のcolumnsをみてみると。
このようになっています。

print(df2.columns)

# MultiIndex([('bw', 1),
#             ('bw', 2),
#             ('bw', 3),
#             ('bw', 4)],
#            names=[None, 'year'])

なので、このように指定すれば列が取り出せます。

print(df2[('bw', 1)])

# id
# 1    50.0
# 2    34.0
# 3    67.0
# Name: (bw, 1), dtype: float64

Comments

[octeast]

columnsをプリントすれば、どのような表現にすればよいかがわかる・・目からうろこです。

[octeast]

ありがとうございます！

Answer 2

今のままでもできないことはありませんが、df2の作り方を変えたほうが楽です。

>>> df
   id  year  bw
0   1     1  50
1   1     2  45
2   1     3  55
3   2     1  34
4   2     2  33
5   3     1  67
6   3     2  66
7   3     3  69
8   3     4  72
>>> df2=df.pivot(values='bw', columns='year', index='id')
>>> df2
year     1     2     3     4
id
1     50.0  45.0  55.0   NaN
2     34.0  33.0   NaN   NaN
3     67.0  66.0  69.0  72.0
>>> df2[1]
id
1    50.0
2    34.0
3    67.0
Name: 1, dtype: float64

Comments

[octeast]

pivotの（）内で、[]が余分ということですね！
ありがとうございます！