python - 填充二维数组以计算两个 linspace 的函数

Question

我有一组想要执行的方程:

x = np.linspace(0, 2, 3)
y = np.linspace(x, x+2, 3)

然后我想用一个计算来填充二维数组:

a = 2*x + y

例如，给定一个数组:

x = [0, 1, 2]

那么，数组y为:

y = [[0, 1, 2],
     [1, 2, 3],
     [2, 3, 4]]

当我执行操作a = 2*x + y时，我应该得到数组:

a = [[0, 1, 2],
     [3, 4, 5],
     [6, 7, 8]]

如何执行此操作，请记住我想对大小高达 10000x10000 (或更大)的数组快速执行此操作？

Answer 1

或者在您的代码中添加两个 T:

print((2*x+y.T).T)

输出:

[[0 1 2]
 [3 4 5]
 [6 7 8]]