NumPy ndarray 的三元运算符?

2024-03-22

NumPy 有三元运算符吗?例如,在 R 中,有一个向量化的if-else功能:

> ifelse(1:10 < 3,"a","b")
 [1] "a" "a" "b" "b" "b" "b" "b" "b" "b" "b"

NumPy 中有类似的东西吗?


您正在寻找numpy.where() https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html:

>>> print numpy.where(numpy.arange(10) < 3, 'a', 'b')
['a', 'a', 'a', 'b', 'b', 'b', 'b', 'b', 'b', 'b']

NumPy 甚至具有泛化功能(将 0、1、2 等映射到值,而不是仅映射 True 和 False):numpy.choose() https://docs.scipy.org/doc/numpy-1.10.0/reference/generated/numpy.choose.html.

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

NumPy ndarray 的三元运算符? 的相关文章

随机推荐