Заполнение двумерного массива для вычисления функции двух линстных пространств

Question

У меня есть этот набор уравнений, который я хочу выполнить:

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

Затем я хочу заполнить 2D-массив вычислением, которое делает:

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 s:

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

Выход:

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