python - 如何将变量步骤插入 Numpy mgrid？

Question

我正在尝试使用 numpy.mgrid 创建两个网格数组，但我想要一种方法来插入一个变量作为步数。

没有可变数量的步骤，numpy.mgrid 使用此代码按预期工作:

x, y = np.mgrid[0:1:3j, 0:2:5j]

但是，我想要的是这样的，因为我无法在生成网格数组的代码行中明确说明我的步骤编号(这些值可能会因脚本中的其他进程而改变):

num_x_steps = 3
num_y_steps = 5
x, y = np.mgrid[0:1:(num_x_steps)j, 0:2:(num_y_steps)j] #Try convert to complex

有没有办法做到这一点，返回与第一种方法等效的结果？

我尝试运行带括号和不带括号的 3 行代码，并尝试了一些其他修改，但似乎没有任何效果。

注意:我尝试阅读 this主题，但我不确定该主题涉及的内容是否适用于此问题；我不太明白问的是什么或者是如何回答的。 (另外，我尝试运行答案中的代码行，它返回了 MemoryError。)如果上述主题确实回答了我的问题，有人可以更好地解释它，以及它如何适用于我的问题吗？

Answer 1

问题在于括号后的 j 不会转换为复数。

In [41]:(1)j
  File "<ipython-input-41-874f75f848c4>", line 1
    (1)j
       ^
  SyntaxError: invalid syntax

将一个值乘以 1j 将起作用，这些行给出的 x, y 等同于您的第一行:

num_x_steps = 3
num_y_steps = 5
x, y = np.mgrid[0:1:(num_x_steps * 1j), 0:2:(num_y_steps * 1j)]