将任意参数集传递给numpy.meshgrid而不进行硬编码

Question

是否可以将一组参数传递给numpy.meshgrid函数而无需输入每个参数？

我希望将类似列表解析的内容传递给此函数，以便不要编写看起来像这样的代码：

mg = numpy.meshgrid(x1,x2,x3,...,xn)

我可以编写一些迭代这些函数的东西并将其提供给函数，就像这个伪代码一样：

mg = numpy.meshgrid([parameter_x for parameter_x in parameter_set])

并产生相同的结果。

我试图研究这个问题并阅读numpy.meshgrid的手册页以寻找无效的线索，我找不到实现此行为的方法。如果有人能指出我可以了解这方面或有解决方案的资源，我将不胜感激。

Answer 1

我找到了问题的答案。

要引用任意集，您可以使用列表推导来创建参数列表，然后使用*前缀访问它，如下所示：

parameter_set = {p1,p2,...,p3}

args = [parameter_x for parameter_x in parameter_set]

mg = numpy.meshgrid(*args)