Итеративно записывайте строки фрейма данных pandas в файл CSV

#python #pandas #csv

Вопрос:

У меня есть CSV-файл, который выглядит так:

 state year candidate forecast_prob result winflag 0 Alabama 2008 Sessions 1.00 Win 1 1 Alabama 2008 Figures 0.00 Lose 0 2 Alabama 2010 Shelby 1.00 Win 1 3 Alabama 2010 Barnes 0.00 Loss 0 4 Alaska 2008 Begich 1.00 Win 1 5 Alaska 2008 Stevens 0.00 Lose 0 6 Alaska 2010 Miller 0.71 Lose 0 7 Alaska 2010 Murkowski 0.21 Win 1 8 Alaska 2010 McAdams 0.08 Loss 0 9 Arizona 2010 Glassman 1.00 Win 1  

Я превратил это в фрейм данных Панд. Как я могу записать каждую строку в другой CSV-файл, где результат = Win ?

Ответ №1:

Вы можете создать новый фрейм данных

new_df = df[df["result"] == "Win"]

И вы можете записать его в csv.

new_df.to_csv("new_df.csv")