numpy append函数

函数原型

numpy.append(arr, values, axis=None)

函数解释

在数组的末尾添加元素，根据数据的形状可以从不同维度进行添加；如果没有指定axis，则数组会展平成一维数组。

函数用法

>>> a = np.zeros((2, 2, 2))
>>> a
array([[[0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.]]])
>>> b = np.zeros((1, 2, 2))
>>> b
array([[[0., 0.],
        [0., 0.]]])
# 没有指定axis
>>> c = np.append(a, b)
>>> c
array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
# 在axis=0上添加
>>> d = np.append(a, b, axis=0)
>>> d
array([[[0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.]]])
>>> e = np.zeros((2, 1, 2))
>>> e
array([[[0., 0.]],

       [[0., 0.]]])
>>> f = np.append(a, e, axis=1)
>>> f
array([[[0., 0.],
        [0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.],
        [0., 0.]]])
>>> g = np.zeros((2, 2, 1))
>>> g
array([[[0.],
        [0.]],

       [[0.],
        [0.]]])
>>> h = np.append(a, g, axis=2)
>>> h
array([[[0., 0., 0.],
        [0., 0., 0.]],

       [[0., 0., 0.],
        [0., 0., 0.]]])