极客教程numpy insert
在numpy中，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)
运行上述代码，输出为：
[ 1 2 10 3 4 5]
在这个示例中，我们向一维数组arr的索引为2的位置插入了值为10的元素。
插入元素到二维数组
接下来我们看看如何向二维数组插入元素。
arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = np.insert(arr, 1, 10, axis=0)
print(new_arr)
运行上述代码，输出为：
[[ 1 2 3]
[10 10 10]
[ 4 5 6]]
在这个示例中，我们向二维数组arr的第一行插入了值为10的元素。
插入多个元素
除了插入单个元素，numpy.insert函数也可以插入多个元素。我们可以将values参数设置为一个数组。
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.insert(arr, [2, 4], [10, 20])
print(new_arr)
运行上述代码，输出为：
[ 1 2 10 3 4 20 5]
在这个示例中，我们向一维数组arr的索引为2和4的位置分别插入了值为10和20的元素。
沿着轴插入元素
在插入多维数组时，我们可以通过axis参数指定沿着哪个轴进行插入。
arr = np.array([[1, 1], [2, 2], [3, 3]])
new_arr = np.insert(arr, 1, 10, axis=1)
print(new_arr)
运行上述代码，输出为：
[[ 1 10 1]
[ 2 10 2]
[ 3 10 3]]
在这个示例中，我们向二维数组arr的第二列插入了值为10的元素。
插入多行和多列
numpy.insert函数还支持一次性插入多行和多列。
arr = np.array([[1, 2], [3, 4], [5, 6]])
new_arr = np.insert(arr, 1, [[7, 8], [9, 10]], axis=0)
print(new_arr)
运行上述代码，输出为：
[[ 1 2]
[ 7 8]
[ 9 10]
[ 3 4]
[ 5 6]]
在这个示例中，我们向二维数组arr的第一行插入了两行数据[7, 8]和[9, 10]。
总结
通过本文的介绍，我们详细讲解了numpy.insert函数的用法，包括向一维数组和多维数组中插入单个元素、多个元素，以及沿着不同轴插入元素。numpy.insert函数的灵活性和功能丰富性使得在数组处理中更加方便和高效。