numpy - 在 Julia 中替代 python 的 np.meshgrid()？

Question

我想知道 Julia 中 NumPy-Python 的 np.meshgrid() 是否有任何替代方案？

#Python 引用:https://numpy.org/doc/stable/reference/generated/numpy.meshgrid.html

Answer 1

请记住，在 Julia 中，您通常可以使用智能广播操作来避免 meshgrid。例如。给定一个对标量值和两个“向量”xs 和 ys 进行操作的函数 f(x, y)，您可以编写 f.(xs, ys') 产生几乎任何结果 meshgrid 会给你和更多:

julia> xs = 0:0.5:1
0.0:0.5:1.0

julia> ys = 0.0:1.0
0.0:1.0:1.0

julia> f(x, y) = (x, y)                # equivalent to meshgrid
f (generic function with 1 method)

julia> f.(xs, ys')
3×2 Matrix{Tuple{Float64, Float64}}:
 (0.0, 0.0)  (0.0, 1.0)
 (0.5, 0.0)  (0.5, 1.0)
 (1.0, 0.0)  (1.0, 1.0)

julia> g(x, y) = x*y                   # more efficient than meshgrid + product
g (generic function with 1 method)

julia> g.(xs, ys')
3×2 Matrix{Float64}:
 0.0  0.0
 0.0  0.5
 0.0  1.0