太多的值来解包试图调用函数的两个数据帧列使用应用和lambda

Question

我正在尝试使用Geopy库更正代码，以获得某些地址的lat long。

所以我写了这个函数：

def latlong(street, city):
    try:
        location = geolocator.geocode(str(street) + ', ' + str(city))
        if location is None:
            return 0, 0
        return location.latitude, location.longitude
    except:
        return 0, 0

df['LAT'], df['LONG'] = df.apply(lambda x: latlong(x['STREET'], x['CITY']), axis = 1)

代码运行了一段时间，然后出现以下错误：

ValueError:要解压缩的值太多（应为2个）

但是，如果我将代码更改为有两个函数，一个用于lat，另一个用于long，它工作得很好，但是运行时间会增加一倍，我将对API执行两倍的请求。

def lat(street, cidade):
    location = geolocator.geocode(str(street) + ', ' + str(city))
    if location is None:
        return 0
    return location.latitude

def long(street, cidade):
    location = geolocator.geocode(str(street) + ', ' + str(city))
    if location is None:
        return 0
    return location.longitude

df['LAT'] = df.apply(lambda x: lat(x['STREET'], x['CITY']), axis = 1)
df['LONG'] = df.apply(lambda x: long(x['STREET'], x['CITY']), axis = 1)

是否有一种方法可以使用apply和lambda调用函数来同时填充数据帧的两列？如何修复此“太多值无法解包”错误？

提前感谢您的帮助！

Answer 1

为了避免进行两次API调用并使程序更快，我首先将两个lat long都添加到一列中。然后将其拆分为两列。这样,，

df['LAT_LOG'] = df.apply(lambda x: latlong(x['STREET'], x['CITY']), axis = 1)
df['LAT'] = df.apply(lambda x: x[0][0], axis=1)
df['LONG'] = df.apply(lambda x: x[0][1], axis=1)

稍后可以删除LAT_LONG列：

del df['LAT_LOG']