将函数元素应用于两个DataFrame

Question

如何将一个函数z_ij = f(x_ij, y_ij)从数据框X和Y同样大小的,并保存结果数据帧Z？

Answer 1

以防万一有人像我一样发现自己在这里，现在有一个函数可以为 pandas 执行此操作！

Z = X.combine(Y, lambda x, y: f(x, y))

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.combine.html

Answer 2

这取决于你,很多功能已经矢量化什么样的功能,用于数据帧,如+-*/等,所以对于这些功能,你可以简单地做Z = X + Y或Z = X - Y等.

对于更通用的函数,您可以使用它numpy.vectorize来制作它的矢量化版本,然后应用于两个数据框:

import numpy as np
import pandas as pd

X = pd.DataFrame([[1,2], [3,4]])
Y = pd.DataFrame([[2,1], [3,3]])
?
def f(x, y):                      # this is a demo function that takes in two ints and 
    return str(x) + str(y)        # concatenate them as str
?
vecF = np.vectorize(f)            # vectorize the function with numpy.vectorize
?
X
#   0   1
#0  1   2
#1  3   4

Y
#   0   1
#0  2   1
#1  3   3

pd.DataFrame(vecF(X, Y))          # apply the function to two data frames

#    0   1
#0  12  21
#1  33  43