物理の駅 Physics station by 現役研究者

テクノロジーは共有されてこそ栄える

Python + pandas でcsvデータの一部を除去して保存する

pandas で読んだデータの一部を除去して保存する方法。下の例ではdata1という変数が1000以下のものを除去している。

import pandas as pd
df = pd.read_csv('run0001.csv')

index_drop = []
for i, row in enumerate(dfdump.itertuples()):
    if row.data1<1000:
        index_drop.append(i)
        continue

df_valid = df.drop(index_drop) #dfは変化しない
df_valid.to_csv('run0001_valid.csv')