julia - 在 Julia 中使用 ndgrid/meshgrid 功能

Question

我试图在 Julia 中找到类似于 MATLAB 的 meshgrid 的功能或 ndgrid .我知道 Julia 已经定义了 ndgrid在 the examples但是当我尝试使用它时，出现以下错误。

UndefVarError: ndgrid not defined

任何人都知道如何获得内置 ndgrid函数来工作，或者可能是我还没有找到的另一个函数或提供这些方法的库(内置函数将是首选)？在这种情况下，我宁愿不写我自己的。

谢谢!

Answer 1

我们更愿意避免使用这些函数，因为它们分配通常不需要的数组。这些数组中的值的结构非常规则，不需要存储；它们只能在迭代期间计算。例如，另一种方法是编写一个数组推导式:

julia> [ 10i + j for i=1:5, j=1:5 ]
5×5 Array{Int64,2}:
 11  12  13  14  15
 21  22  23  24  25
 31  32  33  34  35
 41  42  43  44  45
 51  52  53  54  55

或者，您可以写 for循环，或迭代 product迭代器:

julia> collect(Iterators.product(1:2, 3:4))
2×2 Array{Tuple{Int64,Int64},2}:
 (1, 3)  (1, 4)
 (2, 3)  (2, 4)