使用pyzipcodes从Zip Code列创建States列

Question

我有一个名为df1的数据框架。它包括一个邮政编码列(例如。(包括邮政编码)，我想使用pyzipcodes库创建一个名为state的新列，并将邮政编码映射到各州。有人知道如何调用和使用pyzipcodes来完成这个任务吗?

我能够获得类似于"ZipCode(zip='85711'， city='Tucson'， state='AZ'…"的州名条目。但我只想提取相关的州与邮政编码。

Answer 1

您可以使用pandas.Series.map将邮政编码列与pyzipcode创建的ZipCode对象(这基本上是一个字典)进行映射。

from pyzipcode import ZipCodeDatabase
zcdb = ZipCodeDatabase()
df["state"] = df["zp"].map(lambda x: zcdb[x].state)

输出:

print(df)
zp state
0  85711    AZ