在NumPy数组的末尾添加数值

  • Post category:Python

当我们想在NumPy数组的末尾添加数值时,我们可以使用NumPy中提供的append()函数。

第一步:导入NumPy库

要使用NumPy,我们需要首先导入该库。可以使用以下代码导入NumPy库:

import numpy as np

第二步:创建NumPy数组

创建一个NumPy数组,可以使用以下代码:

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

现在,我们创建了一个包含1、2、3的一维数组a。

第三步:使用append()函数在末尾添加数值

现在,我们想要在NumPy数组的末尾添加元素4,可以使用以下代码:

a = np.append(a, 4)

现在,数组a包含元素1、2、3、4。

示例一:向二维数组中添加一行

现在,让我们来看看如何向二维数组中添加一行。首先,我们需要创建一个二维数组b:

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

现在,我们要添加的行是[7, 8, 9]。我们可以使用以下代码在b的末尾添加一行:

b = np.append(b, [[7, 8, 9]], axis=0)

现在的二维数组b为:

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

其中,axis=0表示添加行,axis=1表示添加列。

示例二:向二维数组中添加一列

现在,让我们来看看如何向二维数组中添加一列。同样是二维数组b,我们要添加的列是[10, 11]。可以使用以下代码在b的末尾添加一列:

b = np.append(b, [[10], [11], [12]], axis=1)

现在的二维数组b为:

array([[ 1,  2,  3, 10],
       [ 4,  5,  6, 11],
       [ 7,  8,  9, 12]])

如果我们要添加多列,可以使用以下代码:

b = np.append(b, np.array([[10, 11, 12],[13, 14, 15],[16, 17, 18]]), axis=1)

现在的二维数组b为:

array([[ 1,  2,  3, 10, 10, 11, 12],
       [ 4,  5,  6, 11, 13, 14, 15],
       [ 7,  8,  9, 12, 16, 17, 18]])

这里,我们使用了np.array()函数创建了一个新的数组,并将其与原数组进行了合并。

通过以上示例,我们可以很好的掌握了如何在NumPy数组的末尾添加数值。