пытается экспортировать данные по городам из CSV в текст

Question

Я пытаюсь экспортировать данные разных лет, сгруппированных по городам, из CSV в текстовый файл с использованием Python, но я не уверен, как это сделать

с CSV-файлом, например

year    rainfall   city
2019      20         A
2019      10         B
2018      18         A
2018       9         B

import pandas as pd #used for other function in program
data = pd.read_csv(file.csv)
...
city=[]
for col in csv.columns:
   if "city" in col.lower():
   citylist = list(csv[col])
      for ct in citylist:
         if ct not in city:
           city.append(ct)

for numcity in city
    textfile= open(file.txt,"w")
    textfile.write()
    textfile.close

The outcome trying to achieve

A.txt
year   rainfall   city
2019      20        A
2018      18        A

B.txt
year   rainfall   city
2019      10        A
2018       9        A

Answer 1

Если вам нужно разделить фрейм данных по городам, вы можете использовать groupby:

for city, grp in data.groupby("city"):
    grp.to_csv(city + ".txt", index=False, sep="\t")