ndarray和异构ndarray之间的操作

Question

我一直在尝试使用一种方法来添加这两个ndarray，其中之一在每行中具有不同数量的元素：

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

我不认为有“ numpy-fast”解决方案。我认为您需要使用for循环遍历a并分别添加每一行。

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