numpy中有功能吗？

Question

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

1. 比较第一和第二系列的值.
2. 如果第一个更大,则从第三个系列中取出元素的弧 - 窦.
3. 否则采用弧余弦.

这是我到目前为止所做的事情:

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

不幸的是,如果不起作用,因为numpy.if不存在.有办法克服这个问题吗？

Answer 1

我想你正在寻找numpy.where:

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

对于1D输入,

where(condition, [x, y])

相当于

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