Pandasで列名を取得する方法（3つの方法）

次のメソッドを使用して、pandas DataFrame の列名を取得できます。

方法 1: すべての列名を取得する

 list (df)

方法 2: 列名をアルファベット順に取得する

 sorted (df)

方法 3: 特定のデータ型の列名を取得する

 list ( df.select_dtypes (include=[' int64 ', ' bool ']))

次の例は、次の pandas DataFrame で各メソッドを使用する方法を示しています。

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F'],
                   ' points ': [18, 22, 19, 14, 14, 11],
                   ' assists ': [5, 7, 7, 9, 12, 9],
                   ' playoffs ': [True, False, False, True, True, True]})

#view DataFrame
print (df)

  team points assists playoffs
0 A 18 5 True
1 B 22 7 False
2 C 19 7 False
3 D 14 9 True
4 E 14 12 True
5 F 11 9 True

例 1: すべての列名を取得する

pandas DataFrame 内のすべての列名を取得する最も簡単な方法は、次のようにlist()を使用することです。

 #get all column names
list (df)

['team', 'points', 'assists', 'playoffs']

結果は、pandas DataFrame の 4 つの列名を含むリストになります。

例 2: 列名をアルファベット順に取得する

pandas DataFrame の列名をアルファベット順に取得するには、次のようにsorted()関数を使用できます。

 #get column names in alphabetical order
sorted (df)

['assists', 'playoffs', 'points', 'team']

結果は、pandas DataFrame の 4 つの列名がアルファベット順にリストされたリストになります。

reverse=True引数を使用して、列名を逆アルファベット順で取得することもできます。

 #get column names in reverse alphabetical order
sorted (df, reverse= True )

['team', 'points', 'playoffs', 'assists']

例 3: 特定のデータ型の列名を取得する

次の構文を使用して、DataFrame の各列のデータ型を表示できます。

 #view data type of each column
df. dtypes

team object
int64 dots
assists int64
playoffs bool
dtype:object

その後、 select_dtypes()関数を使用して、特定のデータ型の列名のみを取得できます。

たとえば、次の構文を使用すると、 int64またはboolデータ型を持つ列名のみを取得できます。

 #get all columns that have data type of int64 or bool
list ( df.select_dtypes (include=[' int64 ', ' bool ']))

['points', 'assists', 'playoffs']

結果は、 int64またはboolデータ型を持つ列名のリストです。

追加リソース

次のチュートリアルでは、パンダで他の一般的な操作を実行する方法を説明します。

パンダ: 列の名前を変更する方法
パンダ: 列をインデックスとして設定する方法
Pandas: 列が値と一致する行のインデックスを取得します