Question

我想添加两个不同形状的numpy数组，但没有广播，而是将“缺失”值视为零。像

这样的例子可能最容易

[1, 2, 3] + [2] -> [3, 2, 3]

或

[1, 2, 3] + [[2], [1]] -> [[3, 2, 3], [1, 0, 0]]

我事先并不知道这些形状。

我正在弄乱每个np.shape的输出，试图找到包含它们的最小形状，将每个包含在该形状的零编辑数组中，然后添加它们。但似乎有很多工作，是否有更简单的方法？

提前致谢！

编辑：通过“大量的工作”我的意思是“为我做了很多工作”而不是机器，我寻求优雅而不是效率：我努力获得最小的形状同时保持它们两者

def pad(a, b) :
    sa, sb = map(np.shape, [a, b])
    N = np.max([len(sa),len(sb)])
    sap, sbp = map(lambda x : x + (1,)*(N-len(x)), [sa, sb])
    sp = np.amax( np.array([ tuple(sap), tuple(sbp) ]), 1)

不漂亮： - /

Answer 1

我正在弄乱每个np.shape的输出，试图找到包含它们的最小形状，将每个包含在该形状的零编辑数组中，然后添加它们。但似乎有很多工作，是否有更简单的方法？

获得np.shape是微不足道的，找到容纳两者的最小形状非常容易，当然添加是微不足道的，所以唯一的“很多工作”部分是“将每个部分嵌入零 - 这个形状的ed数组“。

是的，你可以通过调用resize方法（或resize函数）来消除这种情况，如果你想制作副本而不是就地更改它们。正如文档解释：

扩大数组：...缺少的条目用零填充

例如，如果你静态地知道维度：

>>> a1 = np.array([[1, 2, 3], [4, 5, 6]])
>>> a2 = np.array([[2], [2]])
>>> shape = [max(a.shape[axis] for a in (a1, a2)) for axis in range(2)]
>>> a1.resize(shape)
>>> a2.resize(shape)
>>> print(a1 + a2)
array([[3, 4, 3],
       [4, 5, 6]])

Answer 2

这是我能想到的最好的：

import numpy as np

def magic_add(*args):
    n = max(a.ndim for a in args)
    args = [a.reshape((n - a.ndim)*(1,) + a.shape) for a in args]
    shape = np.max([a.shape for a in args], 0)
    result = np.zeros(shape)

    for a in args:
        idx = tuple(slice(i) for i in a.shape)
        result[idx] += a
    return result

如果您知道对结果有多少维度，可以稍微清理for循环，例如：

for a in args:
    i, j = a.shape
    result[:i, :j] += a

Answer 3

您可以尝试我的解决方案-对于1维数组，您必须将数组扩展为尺寸2（如下例所示），然后再将其传递给函数。

import numpy as np
import timeit


matrix1 = np.array([[0,10],
                    [1,20],
                    [2,30]])
matrix2 = np.array([[0,10],
                    [1,20],
                    [2,30],
                    [3,40]])
matrix3 = np.arange(0,0,dtype=int) # empty numpy-array
matrix3.shape = (0,2) # reshape to 0 rows
matrix4 = np.array([[0,10,100,1000],
                    [1,20,200,2000]])
matrix5 = np.arange(0,4000,1)
matrix5 = np.reshape(matrix5,(4,1000))
matrix6 = np.arange(0.0,4000,0.5)
matrix6 = np.reshape(matrix6,(20,400))
matrix1 = np.array([1,2,3])
matrix1 = np.expand_dims(matrix1, axis=0)
matrix2 = np.array([2,1])
matrix2 = np.expand_dims(matrix2, axis=0)


def add_2d_matrices(m1, m2, pos=(0,0), filler=None):
    """
    Add two 2d matrices of different sizes or shapes,
    offset by xy coordinates, whereat x is "from left to right" (=axis:1)
    and y is "from top to bottom" (=axis:0)
    Parameterse:
        - m1: first matrix
        - m2: second matrix
        - pos: tuple (x,y) containing coordinates for m2 offset,
        - filler: gaps are filled with the value of filler (or zeros)
    Returns:
        - 2d array (float):
            containing filler-values, m1-values, m2-values
            or the sum of m1,m2 (at overlapping areas)
    Author:
        Reinhard Daemon, Austria
    """
    # determine shape of final array:
    _m1 = np.copy(m1)
    _m2 = np.copy(m2)
    x,y = pos
    y1,x1 = _m1.shape
    y2,x2 = _m2.shape
    xmax = max(x1, x2+x)
    ymax = max(y1, y2+y)

    # fill-up _m1 array with zeros:
    y1,x1 = _m1.shape
    diff = xmax - x1
    _z = np.zeros((y1,diff))
    _m1 = np.hstack((_m1,_z))
    y1,x1 = _m1.shape
    diff = ymax - y1
    _z = np.zeros((diff,x1))
    _m1 = np.vstack((_m1,_z))

    # shift _m2 array by 'pos' and fill-up with zeros:
    y2,x2 = _m2.shape
    _z = np.zeros((y2,x))
    _m2 = np.hstack((_z,_m2))
    y2,x2 = _m2.shape
    diff = xmax - x2
    _z = np.zeros((y2,diff))
    _m2 = np.hstack((_m2,_z))
    y2,x2 = _m2.shape
    _z = np.zeros((y,x2))
    _m2 = np.vstack((_z,_m2))
    y2,x2 = _m2.shape
    diff = ymax - y2
    _z = np.zeros((diff,x2))
    _m2 = np.vstack((_m2,_z))

    # add the 2 arrays:
    _m3 = _m1 + _m2

    # find and fill the "unused" positions within the summed array:
    if filler not in (None,0,0.0):
        y1,x1 = m1.shape
        y2,x2 = m2.shape
        x1min = 0
        x1max = x1-1
        y1min = 0
        y1max = y1-1
        x2min = x
        x2max = x + x2-1
        y2min = y
        y2max = y + y2-1
        for xx in range(xmax):
            for yy in range(ymax):
                if x1min <= xx <= x1max and y1min <= yy <= y1max:
                    continue
                if x2min <= xx <= x2max and y2min <= yy <= y2max:
                    continue
                _m3[yy,xx] = filler

    return(_m3)






t1 = timeit.Timer("add_2d_matrices(matrix5, matrix6, pos=(1,1), filler=111.111)", \
"from __main__ import add_2d_matrices,matrix5,matrix6")
print("ran:",t1.timeit(number=10), "milliseconds")

print("\n\n")
my_res = add_2d_matrices(matrix1, matrix2, pos=(1,1), filler=99.99)
print(my_res)

添加不同形状的numpy数组

3 个答案: