python - 根据条件在 Pandas 系列中分配值？

Question

我有一个数据框df，例如

A B
1 2
3 4

然后我想创建 2 个新系列

t = pd.Series()
r = pd.Series()

我能够使用条件 cond 为 t 赋值，如下所示

t = "1+" + df.A.astype(str) + '+' + df.B.astype(str)
cond = df['A']<df['B']
team[cond] = "1+" + df.loc[cond,'B'].astype(str) + '+' + df.loc[cond,'A'].astype(str)

但是我在使用 r 时遇到了问题。我只希望当 con 满足时，r 包含值 2，否则为 1

如果我尝试

r = 1
r[cond] = 2

然后我得到TypeError:'int'对象不支持项目分配

我想我可以通过 df 运行 for 循环，并通过 df 的每一行检查 cond 中的情况，但我是想知道 Pandas 是否提供了更有效的方法？

Answer 1

你会 mock 这是多么容易:

r = cond + 1

原因是 cond 是一个 bool 值(True 和 False)，其计算结果为 1 和 0。如果向其中添加 1，它会将 bool 值强制转换为 int，这意味着 True 映射到 2，False 映射到 1。

df = pd.DataFrame({'A': [1, 3, 4], 
                   'B': [2, 4, 3]})

cond = df['A'] < df['B']

>>> cond + 1
0    2
1    2
2    1
dtype: int64