numpy.insert()的具体使用方法

numpy.insert()函数用于在给定的轴上沿指定的位置插入值。它接受以下参数：

numpy.insert(arr, obj, values, axis=None)

• arr：输入数组
• obj：插入位置
• values：要插入的值
• axis：沿着哪个轴插入，默认为None，表示展开数组

下面是两个示例，演示了如何使用numpy.insert()函数。

示例一：在一维数组中插入值

import numpy as np

# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])

# 在第三个位置插入值
new_arr = np.insert(arr, 2, 10)

# 打印结果
print(new_arr)

在上面的示例中，我们首先创建了一个一维数组arr。然后我们使用numpy.insert()函数在第三个位置插入了值10。最后，我们打印出了新的数组new_arr。

示例二：在二维数组中插入值

import numpy as np

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

# 在第二插入值
new_arr = np.insert(arr, 1, [7, 8], axis=0)

# 打印结果
print(new_arr)

在上面的示例中，我们首先创建了一个二维数组arr。然后我们使用numpy.insert()函数在第二行插入了值[7, 8]。最后，我们打印出了新的数组new_arr`。

总结

numpy.insert()函数是一个非常有用的函数，它可以在给定的轴上沿指定的位置插入值。在使用该函数时，需要注意输入数组的维度和插入位置正确性。