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

Question

我有一个pandas数据框，其中包含列经度和纬度。我想从他们那里得到X和Y。utm中有一个函数，名为from_latlon，用于执行此操作。它接收纬度和经度，并给出[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一次以节省时间。我看了看这里，这里和这里，但是我找不到一种方法来使用一个应用函数制作两列。谢谢。

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))

Answer 2

可以从函数返回一个列表：

d = pandas.DataFrame({
    "A": [1, 2, 3, 4, 5],
    "B": [8, 88, 0, -8, -88]
})

def foo(row):
    return [row["A"]+row["B"], row["A"]-row["B"]]

>>> d.apply(foo, axis=1)
    A   B
0   9  -7
1  90 -86
2   3   3
3  -4  12
4 -83  93

您还可以返回一个序列。这允许您指定返回值的列名：

def foo(row):
    return pandas.Series({"X": row["A"]+row["B"], "Y": row["A"]-row["B"]})

>>> d.apply(foo, axis=1)
    X   Y
0   9  -7
1  90 -86
2   3   3
3  -4  12
4 -83  93