python pandas-将带有两个参数的函数应用于列

Question

你能用两个不同列中的值作为参数创建一个 python pandas 函数吗？

如果两列的值在同一范围内，我有一个返回 1 的函数。否则返回 0：

def segmentMatch(RealTime, ResponseTime):
    if RealTime <= 566 and ResponseTime <= 566:
        matchVar = 1
    elif 566 < RealTime <= 1132 and 566 < ResponseTime <= 1132:
        matchVar = 1
    elif 1132 < RealTime <= 1698 and 1132 < ResponseTime <= 1698:
        matchVar = 1
    else:
        matchVar = 0
    return matchVar

我希望第一个参数 RealTime 成为我的数据框中的一列，这样函数将获取该列中每一行的值。例如RealTime 是 df['TimeCol']，第二个参数是 df['ResponseCol']`。我希望结果是数据框中的一个新列。我遇到了severalthreads，他们回答了类似的问题，但看起来这些参数是变量，而不是数据框行中的值。

我尝试了以下方法，但没有成功：

df['NewCol'] = df.apply(segmentMatch, args=(df['TimeCol'], df['ResponseCol']), axis=1)

Answer 1

执行此操作的chain-friendly 方式是通过assign()：

df.assign( NewCol = lambda x: segmentMatch(x['TimeCol'], x['ResponseCol']) )

Answer 2

如果您在外部定义函数，则实际上并不需要 lambda 函数：

def segmentMatch(vec):
    RealTime = vec[0]
    ResponseTime = vec[1]
    if RealTime <= 566 and ResponseTime <= 566:
        matchVar = 1
    elif 566 < RealTime <= 1132 and 566 < ResponseTime <= 1132:
        matchVar = 1
    elif 1132 < RealTime <= 1698 and 1132 < ResponseTime <= 1698:
        matchVar = 1
    else:
        matchVar = 0
    return matchVar

df['NewCol'] = df[['TimeCol', 'ResponseCol']].apply(segmentMatch, axis=1)

如果“segmentMatch”要返回一个包含 2 个值的向量，您可以执行以下操作：

def segmentMatch(vec):
    ......
    return pd.Series((matchVar1, matchVar2)) 

df[['NewCol', 'NewCol2']] = df[['TimeCol','ResponseCol']].apply(segmentMatch, axis=1)

Answer 3

为什么不这样做呢？

df['NewCol'] = df.apply(lambda x: segmentMatch(x['TimeCol'], x['ResponseCol']), axis=1)

我们现在不是像您的示例那样尝试将列作为参数传递，而是简单地将每行中的适当条目作为参数传递，并将结果存储在'NewCol'。