Как создать файлы CSV на основе значения столбца [дубликат] ⇐ Python

[Anonymous]

У меня есть основной набор данных в pandas usa_df, который выглядит следующим образом

Код: Выделить всё

usa_df = {
"state_name": ["Alabama", "Alabama", "Alaska", "Alaska", "Arkansas"],
"persons": [1, 2, 1, 1, 1],
"day_week": ["monday", "monday", "tuesday", "monday", "tuesday"]
}

В наборе данных около 200 тысяч строк, и каждое название штата отображается в нескольких тысячах строк. Каждая строка представляет собой зарегистрированное преступление, при этом название штата, люди и день недели являются точками данных примерно из ста).
У меня есть код, который фильтрует столбец state_name по конкретное состояние, а затем экспортирует его в формате CSV

Код: Выделить всё

state_df = usa_df[usa_df.state_name == "Alabama"]
state_df.reset_index(drop=True)
os.chdir("/path/to/file")
state_df.to_csv("./Alabama.csv", index=False)

Этот метод работает нормально, но я пытаюсь найти способ, который автоматически создаст файл для каждого состояния без необходимости делать это пятьдесят раз.
Поскольку я новичок, я не знаю, как действовать дальше, кроме, возможно, установки списка состояний в виде списка [str], но кроме этого я заблудился. Можно ли автоматически создать все 50 файлов для каждой строки, соответствующей определенному состоянию? Спасибо за помощь!

Подробнее здесь: https://stackoverflow.com/questions/791 ... lumn-value