pandas DataFrameのカラムヘッダからのリスト取得
pandasのDataFrameからカラムヘッダのリストを取得したいと思います。 DataFrameはユーザーの入力に基づいているので、何列あるのか、何と呼ばれているのかはわかりません。
例えば、次のようなDataFrameが与えられたとします。
>>> my_dataframe
y gdp cap
0 1 2 5
1 2 3 9
2 8 7 2
3 3 4 7
4 6 7 7
5 4 8 3
6 8 2 8
7 9 9 10
8 6 6 4
9 10 10 7
次のようなリストを取得したいと思います。
>>> header_list
['y', 'gdp', 'cap']
904
3
とすることで、値をリストとして取得することができます。
また、単純に:を使うこともできます(Ed Chum's answerに示されています)。
それは、
my_dataframe.columns
として利用できます。