Julia 相当于 numpy 的 where 函数吗?

2024-02-08

在Python中,where在 numpy 中,根据给定条件选择数组中的元素。

>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a < 5, a, 10*a)
array([ 0,  1,  2,  3,  4, 50, 60, 70, 80, 90])

那么朱莉娅呢?filter将被用作选择元素,但它会删除其他元素,如果if未使用表达式。但是,我不想使用if.

我需要编写更复杂的函数吗filter(没有if)或任何其他替代方案?

EDIT:我找到了解决方案,但如果有人对此有更好的想法,请回答这个问题。

julia > a = collect(1:10)
10-element Array{Int64,1}:
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10

julia> cond = a .< 5
10-element BitArray{1}:
  true
  true
  true
  true
 false
 false
 false
 false
 false
 false

julia> Int.(cond) .* a + Int.(.!cond) .* (10 .* a)
10-element Array{Int64,1}:
   1
   2
   3
   4
  50
  60
  70
  80
  90
 100

有多种方式,最明显的是广播ifelse像这样:

julia> a = 0:9  # don't use collect
0:9

julia> ifelse.(a .< 5, a, 10 .* a)
10-element Array{Int64,1}:
  0
  1
  2
  3
  4
 50
 60
 70
 80
 90

您还可以使用@.宏以确保您得到正确的点:

@. ifelse(a < 5, a, 10a)

或使用理解

[ifelse(x<5, x, 10x) for x in a]

当然,您也可以使用循环。

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

Julia 相当于 numpy 的 where 函数吗? 的相关文章

随机推荐