将函数应用于pandas dataframe的两列以获取两个新列

Question

我有一个pandas数据框，其中有Longitude和Latitude列。我想从他们那里得到X和Y。在utm中有一个名为from_latlon的函数来执行此操作。它接收Latitude和Longitude，并给出[X,Y]。我做的是：

    def get_X(row):
        return utm.from_latlon(row['Latitude'], row['Longitude'])[0]

    def get_Y(row):
        return utm.from_latlon(row['Latitude'], row['Longitude'])[1] 

    df['X'] = df.apply(get_X, axis=1)
    df['Y'] = df.apply(get_Y, axis=1)

我想定义一个函数get_XY，并只应用from_latlon一次以节省时间。我看了一下here、here和here，但是我找不到一种方法来用一个apply函数生成两列。谢谢。

Answer 1

我从一个类似的线程中合并了几个答案，现在有一个通用的多列输入、多列输出模板，我在Jupyter/pandas中使用：

# plain old function doesn't know about rows/columns, it just does its job.
def my_func(arg1,arg2):
    return arg1+arg2, arg1-arg2  # return multiple responses

df['sum'],df['difference'] = zip(*df.apply(lambda x: my_func(x['first'],x['second']),axis=1))