主页 / coding / 问题 / 79224958
Accepted
Ben
Asked: 2024-11-26 08:30:40 +0800 CST

从标量和矩阵形成元素列表

  • 772

我有一个零维numpy标量s和一个二维numpy矩阵m。我想形成一个向量矩阵，其中的所有元素都m与配对，s如下例所示：

import numpy as np

s = np.asarray(5)

m = np.asarray([[1,2],[3,4]])

# Result should be as follows

array([[[5, 1],
        [5, 2]],

       [[5, 3],
        [5, 4]]])

换句话说，我想np.asarray([s, m])在 的最低级别上逐元素地矢量化操作。对于内的m任何多维数组，是否有一种明显的方法来做到这一点？mnumpy

我确信这个在某个地方，但我无法用语言表达，也找不到它。如果你能找到它，请随时将我重定向到那里。

numpy

2 个回答

  1. Best Answer

    一种可能的解决方案是使用broadcast_to和stack函数将两个数组s和m组合成一个沿新轴的数组。步骤如下：

    • 首先，np.broadcast_to(s, m.shape)扩展数组的形状s以匹配数组的形状，m而无需复制数据。

    • 然后，np.stack([np.broadcast_to(s, m.shape), m], axis=-1)加入广播s并m沿着新的最后一个轴

    np.stack([np.broadcast_to(s, m.shape), m], axis=-1)

    输出：

    array([[[5, 1],
        [5, 2]],

       [[5, 3],
        [5, 4]]])
    • 2

  2. 没stack问题。但由于s需要重复，一个好的替代方案是创建所需形状的零数组，并允许broadcasting填充列：

    In [2]: s = np.asarray(5)
   ...: 
   ...: m = np.asarray([[1,2],[3,4]])

In [3]: res = np.zeros((m.shape)+(2,)); res
Out[3]: 
array([[[0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.]]])

In [4]: res[:,:,0] = s; res[:,:,1] = m

In [5]: res
Out[5]: 
array([[[5., 1.],
        [5., 2.]],

       [[5., 3.],
        [5., 4.]]])
    • 0

相关问题