填充二维数组以计算两个线性空间的函数

Question

I have this set of equations I want to perform:

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

I then want to populate the 2D array with a calculation that does:

a = 2*x + y

So for example, given an array:

x = [0, 1, 2]

Then, the array y is:

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

When I perform the operation a = 2*x + y I should get the array:

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

How do I do this, keeping in mind I want to perform this operation quickly for array of size up to 10000x10000 (or larger)?

Answer 1

Or do your code adding two T s:

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

Output:

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