从Pandas[duplicate]中前两列上的函数创建第三列

Question

我有一个功能：

def lookup(group, name):
    try:
        #sql query with group and name
        return answer
    except:
        return 'NULL'

以及数据帧：

+------------+------+
|   group    | name |
+------------+------+
| redteam    | paul |
| blueteam   | pat  |
| yellowteam | mike |
+------------+------+

我想将每行的组和名称输入到我的“lookup”函数中，并将该行的答案作为第三列中的单元格返回：“value”

我已经看了这个和这个问题。但他们不完全是我在做的。

我也研究过这个问题。这就是我正在做的。但对我来说也没有成功。

我试过这个：df['value']=lookup（df[0]，df[1]）
和这个：
df['value']=df.apply（lambda x： lookup（x[0]，x[1]），轴=1）

但它只会使值列全部为“NULL”

这里的任何建议都非常感谢。

Answer 1

如果没有一个好的可重复的示例，很难进行测试，但您可以尝试这样做，假设group不是数据帧的索引：

以下是一个完整的示例：

df = pd.DataFrame(np.random.rand(10,2), columns=['group', 'name'])

def lookup(group, name):
    if group > 0.4:
        return 'YES'
    else:
        return 'NULL'

df['value'] = df.apply(lambda x: lookup(x['group'], x['name']), axis=1)

df
      group      name value
0  0.088522  0.221607  NULL
1  0.366478  0.612860  NULL
2  0.018939  0.995080  NULL
3  0.143422  0.590115  NULL
4  0.747373  0.888054   YES
5  0.960380  0.586448   YES
6  0.671776  0.151144   YES
7  0.632334  0.802551   YES
8  0.041953  0.387241  NULL
9  0.557183  0.199470   YES