使用 City 和 State 列查找纬度和经度

Question

我有一列包含“CITY”和“STATE”字符串。我尝试使用 geopy 库中的地理编码器来计算纬度和经度，但它超时，因为它超过了请求的数量，而且数据集中大约有 85895 行。所以我对“CITY_STATE”列进行了计数，有 1340 个值。有没有办法可以找到这 1340 个值的纬度和经度，然后将这些值映射到相应的“CITY”和“STATE”。

from opencage.geocoder import OpenCageGeocode
key = 'my_key'  # get api key from : https://opencagedata.com
geocoder = OpenCageGeocode(key)
list_lat = []   # create empty lists

list_long = []
for index, row in train.iterrows(): # iterate over rows in dataframe



City = row['CITY']
State = row['STATE']       
query = str(City)+','+str(State)

results = geocoder.geocode(query)   
lat = results[0]['geometry']['lat']
long = results[0]['geometry']['lng']

list_lat.append(lat)
list_long.append(long)

# create new columns from lists    

train['lat'] = list_lat   

test['lon'] = list_long

Answer 1

我认为您可以将GroupBy.apply 与自定义函数一起使用，并将这两列加入系列s：

s = train['CITY'].astype(str) + ', ' + train['STATE'].astype(str)

def func(x):
    results = geocoder.geocode(x.name) 
    x['lat'] = results[0]['geometry']['lat']
    x['lon'] = results[0]['geometry']['lng']
    return x

train = train.groupby(s).apply(func)