numpy数组可以包含本身为numpy数组的元素吗？

Question

我对python不太熟悉，所以我提前道歉。 是否有可能有一个numpy数组，例如numpy.array([a, b, c])并将该数组作为元素添加到空numpy数组中？

假设这是可能的，则可以对主numpy数组中多个数组的每个元素的第一个值求和。 例如，

numpy.array([numpy.array([a,b,c]), numpy.array([d,e,f])])

然后成为

numpy.array([a + d, b + e, c + f])

希望我能清楚地解释一下，如果不确定，请随时要求我扩展。

非常感谢 ：-）

Answer 1

数组的数组本质上就是矩阵，所以我只做一个n维的numpy数组。 这样，您可以在给定方向上逐个元素地求和。

import numpy as np
x = np.array([[1, 2, 3], [4, 5, 6]], np.int32) # 2x3 matrix

通常，您可能希望遍历索引，但对于您的示例，单独进行求和则更有意义：)

a_plus_d = sum(x[:,0]) # sum first column
b_plus_e = sum(x[:,1]) # sum second column
c_plus_f = sum(x[:,2]) # sum third column

Answer 2

2个1d阵列：

In [79]: x1=np.array([1,2,3])
In [80]: x2=np.array([4,5,6])

制作新数组-2d，带有2行

In [81]: x12 = np.array((x1,x2))
In [82]: x12
Out[82]: 
array([[1, 2, 3],
       [4, 5, 6]])

np.array([[1,2,3],[4,5,6]])做同样的事情。

可以逐个元素地对数组求和：

In [83]: x1 + x2
Out[83]: array([5, 7, 9])

2d数组的行也可以求和：

In [84]: x12.sum(axis=0)
Out[84]: array([5, 7, 9])