python - numpy 中是否有通用的 if 函数？

Question

我有三个系列。我需要按元素执行以下操作:

1. 比较第一个和第二个系列的值。
2. 如果第一个较大，取第三个系列元素的反正弦。
3. 否则取反余弦。

到目前为止，这是我设法做到的:

numpy.if(numpy.less(s1,s2),numpy.arcsin(s3),numpy.arccos(s3))

不幸的是 if 不起作用，因为 numpy.if 不存在。有没有办法克服这个问题？

Answer 1

我想你在找numpy.where :

np.where(s1<s2, np.arcsin(s3), np.arccos(s3))

---

对于一维输入，

where(condition, [x, y])

相当于

[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]