Question

我正在使用美国所有县的covid计数数据框。我想出了如何隔离一个县并将结果导出到csv的方法：

import pandas as pd
covid = pd.read_csv('https://raw.githubusercontent.com/nytimes/covid-19-data/master/us-counties.csv')
agh = covid[covid['county'] == 'Allegheny']
agh.to_csv('AlleghenyCovid.csv')

现在我要创建一个这样的县列表：

countyList = covid.county.unique()

并遍历它们为每个创建一个csv。那就是我被困住的地方。如何使用已知值列表遍历数据框并从每次迭代创建新的数据框？我一直在想类似的东西：

for i in countyList:
    if covid['county'] == i:
        ...

但是给出了一个模棱两可的值错误。我不确定确切需要定义什么。

Answer 1

解决方案迭代county列的唯一列表：

for name in covid.county.unique()
    covid.loc[covid.county == name,:].to_csv(name+'.csv')

对于name所命名的每个县：

我们正在从数据帧covid中选择行，其中county等于name
然后将这样的选择保存到名为name + .csv的CSV文件中。

熊猫：遍历列表以匹配数据框中的值

1 个答案: