如果基于某些其他列不满足条件，则使用值列表填充 Dataframe 列

我有一个看起来像这样的数据框 -

col_1   |   col_2
-------------------
"red"   |    21
-------------------
"blue"  |    31
-------------------
"red"   |    12
-------------------
"blue"  |    99
-------------------
"blue"  |    102

我也有一个像这样的值列表 label = [1,3,2]

我想构建第三列col_3，如果颜色是“红色”，则应该有“是” ，否则在相应的行上应该有 1,3,2。基本上，如果颜色为“蓝色” ，则标签中的值应该一个接一个。col_1

预期输出-

col_1   |   col_2    | col_3
---------------------------
"red"   |    21      |  "Yes"
-----------------------------
"blue"  |    31      |  "1"
------------------------------
"red"   |    12      | "Yes"
------------------------------
"blue"  |    99      |  "3"
------------------------------
"blue"  |    102     |  "2"

我的方法-

我试过np.where()像这样估算使用

np.where(df["col_1"]=="red","Yes",labels)

，但是

ValueError: operands could not be broadcast together with shapes

我相信这是由于df和labels(5 vs 3)的大小不同。

任何人都可以帮助我吗？

谢谢

编辑：

1. 添加了预期输出
2. 在我的方法演示中犯了一些错误，更正了。