逐行插入2D Numpy数组的值

发布于 2021-01-29 15:05:18

我有两个numpy数组:

x = np.array([1,2,3,4,5])
y = np.array([10,20,30,40,50])

我试图得到的是这样的:

array([[  1.  ,   3.25,   5.5 ,   7.75,  10.  ],
       [  2.  ,   6.5 ,  11.  ,  15.5 ,  20.  ],
       [  3.  ,   9.75,  16.5 ,  23.25,  30.  ],
       [  4.  ,  13.  ,  22.  ,  31.  ,  40.  ],
       [  5.  ,  16.25,  27.5 ,  38.75,  50.  ]])

我的方法不是很像Numpy,需要针对非常大的数组进行改进(摆脱for循环):

myarray = np.zeros((5,5))
for idx in np.arange(5):
    myarray[idx,:] = np.linspace(x[idx], y[idx], 5)

在Numpy中执行此操作的最佳方法是什么?这样生成myarray然后对其进行操作会更好吗?

myarray = np.zeros((5,5))
myarray[:,0] = x
myarray[:,-1] = y

array([[  1.,   0.,   0.,   0.,  10.],
       [  2.,   0.,   0.,   0.,  20.],
       [  3.,   0.,   0.,   0.,  30.],
       [  4.,   0.,   0.,   0.,  40.],
       [  5.,   0.,   0.,   0.,  50.]])
关注者
0
被浏览
88
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    这种广播技巧:

    >>> x = np.array([1,2,3,4,5])
    >>> y = np.array([10,20,30,40,50])
    >>> z = np.linspace(0, 1, 5)
    >>> z[None, ...] * (y[..., None] - x[..., None]) + ( x[..., None])
    array([[  1.  ,   3.25,   5.5 ,   7.75,  10.  ],
           [  2.  ,   6.5 ,  11.  ,  15.5 ,  20.  ],
           [  3.  ,   9.75,  16.5 ,  23.25,  30.  ],
           [  4.  ,  13.  ,  22.  ,  31.  ,  40.  ],
           [  5.  ,  16.25,  27.5 ,  38.75,  50.  ]])
    >>>
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看