填充2D数组以计算两个linspace的函数

时间:2019-01-13 07:53:36

标签: python arrays numpy

我要执行以下这组方程:

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(或更大)的数组快速执行此操作?

1 个答案:

答案 0 :(得分:1)

或者您的代码添加两个T

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

输出:

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