在lambda函数中连接两列

Question

我有一个数据框，有两列“first_name”和“name”

我想做的是检查名字的长度，如果是<=3然后创建一个新列“new_name”，并连接“first_name”和“name”列。如果长度为>；3那么新的_名称应该与“first_name”相同

例如：

first_name   name      new_name
 V           David      VDavid

我试着这样做：

df = df.assign(new_name= lambda x: x.first_name + x.name if len(x.first_name) <= 3 else x.first_name)

这段代码应该可以做到这一点，但它没有按预期工作。它只是返回第一个_名称，而不考虑单词的长度

有人能帮我吗

Answer 1

使用：

df = df.assign(new_name= lambda x: np.where(x.first_name.str.len() <= 3, 
                                            x.first_name + x.name, 
                                            x.first_name))