Pandas カラム名の部分一致検索

スポンサーリンク

カラムの多いデータフレームを扱っていると、
df.columnsだと見切れてしまったり、見切れなくても出力が多すぎて探すのがめんどくさいときがある。
「○○売上」みたいなあの列名加工したいんだけどなんだっけ?みたいな。
そういう時用の2つのソリューションを紹介

スポンサーリンク

.contains

df.columns[df.columns.str.contains('hogehoge')]

.filter()

df.filter(like='hogehoge').columns.tolist()

どちらか使えば部分一致しているカラム名をlistで取ってこれる。
個人的にはfilter()が好き。
短いし、書き方もより直感的。

タイトルとURLをコピーしました