PandasにおけるDataFrameの全ての列ヘッダーをリストとして取得する

今回は、PandasにおけるDataFrameの全ての列ヘッダーをリストとして取得してみます。

■Python

今回のPythonのバージョンは、「3.8.5」を使用しています。（Windows10）（pythonランチャーでの確認）

■DataFrameを作成する

DataFrameの全ての列ヘッダーをリストを取得する前に、DataFrameを作成します。

■コード

import pandas as pd

data = {
    '名前':['田中','戸田','林田','森岡','鈴木'],
    '出身':['兵庫','大阪','京都','東京','沖縄'],
    '血液型':['A','B','O','AB','B'],
    '年齢':[35,29,42,56,27]
}

df = pd.DataFrame(data)

print(df)

インポートでPandasモジュールを呼び出します。dataという変数を作成し、その中に「名前」、「出身」、「血液型」、「年齢」の4つの列を格納します。

格納後、dfという変数を作成し、pd.DataFrame()と記述し、DataFrameを作成。作成後、dfという変数に格納します。

■実行

このスクリプトを実行すると、作成したDataFrameが出力されます。

■DataFrameの全ての列ヘッダーをリストとして取得する

DataFrameの用意ができましたので、DataFrameの全ての列ヘッダーをリストとして取得するスクリプトを書いていきます。

■コード

import pandas as pd

data = {
    '名前':['田中','戸田','林田','森岡','鈴木'],
    '出身':['兵庫','大阪','京都','東京','沖縄'],
    '血液型':['A','B','O','AB','B'],
    '年齢':[35,29,42,56,27]
}

df = pd.DataFrame(data)

print(df)

print(list(df.columns.values))

DataFrameの全ての列ヘッダーをリストとして取得する場合は、dfという変数を作成し、pd.DataFrame()と記述し、DataFrameを作成。その後にdf.columns.valuesで列ヘッダーの情報を取得し、その後、list()関数を使用し、リストに変換します。変換したものをprint関数で出力します。

■実行

このスクリプトを「df_headers_2.py」という名前で保存し、コマンドプロンプトから実行してみます。

実行してみると、作成したDataFrame内の列ヘッダーの情報がリスト化されて出力できました。