Pandas DataFrame から列ヘッダーを取得する

Pandas DataFrame は、効率的なデータ操作と分析を可能にする多用途のデータ構造です。一般的なタスクの 1 つは、列ヘッダーの抽出です。これは、DataFrame の構造の概要を取得したり、さらなる処理を行うのに役立ちます。

列の数と名前が不明な、ユーザー入力からインポートされた DataFrame があるとします。 。列ヘッダーをリストとして抽出するには、次のメソッドを利用できます。

方法 1: DataFrame.columns.values を使用する

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns は、列ヘッダーを含む Index オブジェクトを返します。このインデックスを値を使用してリストに変換すると、列名を表す文字列のリストが得られます。

方法 2: DataFrame を使用する.columns

column_headers = list(my_dataframe)

このメソッドは、my_dataframe.columns.values の短縮記法です。列をリストに変換します。

使用例

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

いずれかの方法を使用すると、次の列ヘッダーのリストが取得されます:

['y', 'gdp', 'cap']