如何使用pandas assign创建具有多个条件的新列？

Question

我有一个包含以下列的pandas数据框：

Date, Variable 1, rank_pct

我正在尝试使用assign来创建一个名为＆＃34; ls＆＃34;的新列。其中包含1，-1或0。

1. 如果rank_pct <0.10，则在ls
列中加1
2. 如果rank_pct> 0.90，则在ls。
列中加上-1
3. 如果rank_pct是其他任何内容，请在列中添加0。

我不想使用for循环来遍历rank_pct列中的每一行，我已经尝试过以下内容（它返回一个True / False列：

data = data.assign(ls= data['rank_pct']<.10)

我可以在括号内放置任何命令将T / F转换为1，-1和0吗？

Answer 1

您可以使用嵌套的np.where：

data.assign(ls=np.where(data.rank_pct<.1, 1,
                        np.where(data.rank_pct>.9, -1, 0)))