Операция между ndarray и гетерогенным ndarray

Question

Я пытался найти способ добавить эти два ndarrays, один из которых с разным количеством элементов в каждой строке:

a = np.array([np.array([0, 1]), np.array([4, 5, 6])])
z = np.zeros((3,3))

Ожидаемый результат:

array([[0., 1., 0.],
       [4., 5., 6.]])

Кто-нибудь может придумать способ сделать это, используя numpy?

Answer 1

Я не думаю, что для этого есть «быстрое решение».Я думаю, вам нужно перебрать a с помощью цикла for и добавить каждую строку отдельно.

for i in range(len(a)):
    z[i,:len(a[i])] = z[i,:len(a[i])] + a[i]