numpy m x n 组合网格与 m 和 n 数组-解网

问：

我想知道是否有一种简单的方法可以做到这一点：

a = 
[[1,2],[3,4]]

b =
[[5,6],[7,8],[9,10]]

变成这样：

c = 
    [[[1,2], [1,2], [1,2]],
    [[3,4], [3,4], [3,4]]]
d = 
    [[[5,6], [7,8], [9,10]],
    [[5,6], [7,8], [9,10]]]

所以我可以这样做：

c - d

我已经尝试过使用 np.meshgrid，但它真的很笨拙：

indexes_b, indexes_a = np.meshgrid(
    np.arange(a.shape[0]),
    np.arange(b.shape[0])
)

c = a[indexes_a]
d = b[indexes_b]
c - d # works

python numpy 矩阵 nm

评论

0赞 Quang Hoang 2/3/2023

np.subtract.outer(a,b)或a[None]-b[:,None]

答：

0赞 Jônathan Dambros 2/3/2023 #1

试试这个：

a = [[1,2],[3,4]]
b = [[5,6],[7,8],[9,10]]

c = [[li]*len(b[0]) for li in a]
d = [[li]*len(a) for li in b]

print(c)
print(d)

评论

0赞 Community 2/5/2023

您的答案可以通过额外的支持信息得到改进。请编辑以添加更多详细信息，例如引文或文档，以便其他人可以确认您的答案是正确的。您可以在帮助中心找到有关如何写出好答案的更多信息。

4赞 Corralien 2/3/2023 #2

用：broadcasting

>>> a[:, None] - b
array([[[-4, -4],
        [-6, -6],
        [-8, -8]],

       [[-2, -2],
        [-4, -4],
        [-6, -6]]])

>>> c - d
array([[[-4, -4],
        [-6, -6],
        [-8, -8]],

       [[-2, -2],
        [-4, -4],
        [-6, -6]]])

评论

0赞 mozway 2/3/2023

我也在计算实数，以防万一也需要：cdc = np.tile(a[:,None], (1,b.shape[0],1)) ; d = np.tile(b[None], (a.shape[0],1,1))

上一个：简单的结构增加 C++ Android 二进制文件的大小

下一个：'nm' 和 'objdump' 未知美元符号