【Python】データフレームのカラム名単位でループ処理【pandas】 | ムノログ 合同会社ムジンケイカクプロのITノウハウブログ
Pythonと機械学習

【Python】データフレームのカラム名単位でループ処理【pandas】

python Pythonと機械学習

pandasデータフレームのカラム名を確認。
リストにある名前のカラムであれば、名前表示後の削除。

drop_col = ['TransactionDT', 'V1']

def dc(df)
   for i in df.columns:
       if i in drop_col:
           print(type(i))
           print(i)
           print('======\n')
           df.drop(i, axis=1)
       else:
           print('NO')
dc(df)

何故作ったか忘れました。