将数组馈送到 numpy meshgrid

Question

meshgrid 的编码方式似乎使得网格网格的每个维度都必须单独命名（例如 x 和 y）

xv, yv = np.meshgrid(x, y)

我想知道我是否可以只提供一个包含 x 和 y 的数组来做同样的事情，可能设置一些 meshgrid 选项来指示它按照我需要的方式使用数组。

我想这样做是因为当必须在其上构建网格网格的维度数量很大或先验未知时，它会让生活变得更轻松。

所以问题是

np.meshgrid([1,2],[10,20])

给予

[array([[1, 2],
        [1, 2]]), array([[10, 10],
        [20, 20]])]

而我想要类似的东西

np.meshgrid([[1,2],[10,20]])

但这给出了一个无用的一维数组。

我在doc 中找不到任何有用的选项

Answer 1

这是解包参数列表的情况，您在python中为此使用*-operator：

lst = [[1,2],[10,20]]
np.meshgrid(*lst)

#[array([[1, 2],
#        [1, 2]]), array([[10, 10],
#        [20, 20]])]