Pandas の DataFrame
から、数値タイプ(int64
や float64
)のカラムだけを抽出するには、df.select_dtypes()
メソッドを使用します。
次の例では、サンプルの DataFrame
から数値タイプのカラム名を抽出して列挙しています。
import numpy as np
import pandas as pd
# サンプルの DataFrame を作成
data = {
'列1': [1, 2, 3],
'列2': [1.1, 2.2, 3.3],
'列3': ['A', 'B', 'C']
}
df = pd.DataFrame(data)
# 数値タイプのカラムを抽出してカラム名のリストを取得
numerical_columns = df.select_dtypes(include=[np.number]).columns
# 結果表示
print(numerical_columns)
for col in numerical_columns:
print(col)
Index(['列1', '列2'], dtype='object')
列1
列2
なお、上の例では数値型として np.number
を指定していますが、"number"
という文字列でも大丈夫です。