Question

我有一个3D矩阵a，如下所示：

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

现在，我想在此数组中添加另一个元素，如[[5,6],[6,7]]。所以，新数组将是：

[
 [[1, 2],[2, 3]]
 [[3, 4],[4, 5]]
 [[5, 6],[6, 7]]
]
a.shape
(3, 2, 2)

最好的方法是什么？（我正在处理大数据集，所以我需要最好的方法）

Answer 1

通过添加新轴作为第一个轴3D，将第二个数组扩展到None/np.newaxis后，使用np.vstack进行垂直堆叠，就像这样 -

np.vstack((a,b[None]))

示例运行 -

In [403]: a
Out[403]: 
array([[[1, 2],
        [2, 3]],

       [[3, 4],
        [4, 5]]])

In [404]: b
Out[404]: 
array([[5, 6],
       [6, 7]])

In [405]: np.vstack((a,b[None]))
Out[405]: 
array([[[1, 2],
        [2, 3]],

       [[3, 4],
        [4, 5]],

       [[5, 6],
        [6, 7]]])

Answer 2

您可以使用np.append附加到矩阵：

a = np.array([[[1,2],[2,3]],[[3,4],[4,5]]])
a = np.append(a, [[[5,6],[6,7]]], axis=0)

请注意，我必须在第二部分周围添加一组额外的括号，以使尺寸正确。此外，您必须使用axis，否则它将全部展平为线性阵列。

Answer 3

尝试numpy.append

import numpy as np
a=np.array([[[1,2],[2,3]],[[3,4],[4,5]]])
b=np.array([[3,4],[4,5]])
np.append(a,[b[:,:]],axis=0)

如何在python中将2D矩阵添加到另一个3D矩阵？

3 个答案: