修改 meshgrid 函数

Question

我有一段关于 meshgrid 函数的代码，我想修改它的输出：

x_list = list(range(5))
y_list = list(range(2))
X, Y = meshgrid(x_list, y_list)

打印出：

X
array([[0, 1, 2, 3, 4],
       [0, 1, 2, 3, 4]])

和

    Y = array([[0, 0, 0, 0, 0],
               [1, 1, 1, 1, 1]])

如何修改我的输出以生成

    X = array([0,0,0,0,0],
              [1,1,1,1,1]

和 Y = array([0,1], [0,1], [0,1], [0,1], [0,1]]

相反？我的想法是这样的，使 X 基本上代表 x_axis 值，Y 代表 Y 轴值：

    03 13 23 33 43
    02 12 22 32 42
    01 11 21 31 41
    00 10 20 30 40

最后我做错了什么我知道，但我不知道如何更改 meshgrid 中的代码以获得正常的 x-y 平面。

Answer 1

size= 4
X = np.array([np.zeros(size), np.ones(size)])
Y = X.copy().T

会得到你想要的输出。但是，如果您正在使用 meshgrid，那么您是否提供了期望格式 meshgrid 产生的函数？

Modifying meshgrid function