Добавление массивов numpy определенным образом

Question

У меня есть три массива numpy.

import numpy as np 
    a=np.array((1,2,3,4)) 
    b=np.array((2,3,4,5)) 
    c=np.array((3,4,5,6)) 

, и я хочу, чтобы добавить эти массивы в пути

a : [1 2 3 4] 
+ b : [2 3 4 5] 
+ c :  [ 3 4 5 6] 
    = [1 4 9 12 10 6] 

Есть ли простой способ?

Answer 1

Один интуитивный способ (хотя неэффективное) было бы создать промежуточный массив нулей, а затем заполнить в каждой строке со значениями в a, b, c, ... с соответствующим сдвигом столбца. Если у вас есть этот массив, вы можете просто просуммировать строки, чтобы получить результат, который вы хотите:

import numpy as np 

a = np.array((1,2,3,4)) 
b = np.array((2,3,4,5)) 
c = np.array((3,4,5,6)) 

arr_list = [a, b, c] 

nr = len(arr_list) 
row_lengths = [len(aa) for aa in arr_list] 
nc = np.max(row_lengths) + nr - 1 

digits = np.zeros((nr, nc), a.dtype) 

for ii, aa in enumerate(arr_list): 
     digits[ii, ii:(ii + row_lengths[ii])] = aa 

print(digits) 
# [[1 2 3 4 0 0] 
# [0 2 3 4 5 0] 
# [0 0 3 4 5 6]] 

print(digits.sum(0)) 
# [ 1 4 9 12 10 6] 

Более быстрый метод будет вычислять выход непосредственно, без создания промежуточного 2D массива

out = np.zeros(np.max(row_lengths) + nr - 1, a.dtype) 

for ii, aa in enumerate(arr_list): 
    out[ii:(ii + row_lengths[ii])] += aa 

print(out) 
# [ 1 4 9 12 10 6]