对python numpy.array插入一行或一列的方法详解

  • Post category:Python

下面是关于“对Python NumPy数组插入一行或一列的方法详解”的完整攻略,包含了两个示例。

插入一行

下面是一个示例,演示如何在二维数组中插入一行。

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 创建一个新的一维数组
b = np.array([10, 11, 12])

# 在二维数组中插入一行
c = np.insert(a, 1, b, axis=0)

print(c)

在上面的示例中,我们首先创建了一个二维数组a,然后创建了一个新的一维数组b。使用np.insert()函数将新的一维数组b插入到二维数组a的第1行,得到新的二维数组c。输出结果为:

[[ 1  2  3]
 [10 11 12]
 [ 4  5  6]
 [ 7  8  9]]

需要注意的是,np.insert()函数的第一个参数是要插入的数组,第二个参数是要插入的位置,第三个参数是要插入的值,第四个参数是要插入的轴。在上面的示例中,我们使用axis=0将新的一维数组b插入到二维数组a的第1行。

插入一列

下面是一个示例,演示如何在二维数组中插入一列。

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 创建一个新的一维数组
b = np.array([10, 11, 12])

# 在二维数组中插入一列
c = np.insert(a, 1, b, axis=1)

print(c)

在上面的示例中,我们首先创建了一个二维数组a,然后创建了一个新的一维数组b。使用np.insert()函数将新的一维数组b插入到二维数组a的第1列,得到新的二维数组c。输出结果为:

[[ 1 10  2  3]
 [ 4 11  5  6]
 [ 7 12  8  9]]

需要注意的是,np.insert()函数的第一个参数是要插入的数组,第二个参数是要插入的位置,第三个参数是要插入的值,第四个参数是要插入的轴。在上面的示例中,我们使用axis=1将新的一维数组b插入到二维数组a的第1列。

总结

本文介绍了如何在Python NumPy数组中插入一行或一列。使用np.insert()函数可以在指定位置插入新的一行或一列。需要注意的是,np.insert()函数的第一个参数是要插入的数组,第二个参数是要插入的位置,第三个参数是要插入的值,第四个参数是要插入的轴。