如果不是numpy中的方形matirx,列的总和?

时间:2016-11-01 09:27:54

标签: python numpy matrix jupyter-notebook

a=np.array([[1,2,3],[1]])
print(np.sum(a,axis=1))

我假设矩阵就像

[1,2,3]
[1]

1 个答案:

答案 0 :(得分:0)

不是很一般,因为它仅在第一行最长的情况下才有效,但您可以简单地将数组固定为零数组。 例如:

a = np.array([[1,2,3],[4,5]])

dim_x = len(a[0])
dim_y = a.shape[0]

a0 = np.zeros((dim_y, dim_x))

for i in range(dim_y):
    dim = len(a[i])
    a0[i,0:dim] = a[i]