numpy中数组怎么插入数据

numpy中数组怎么插入数据

在numpy中，数组是用来存储相同类型的数据元素的集合。但有时候我们需要在数组中插入新的数据，比如在数组的特定位置插入一个新的元素。本文将详细介绍numpy中数组如何插入数据的方法。

使用numpy.insert()函数插入数据

numpy中的insert()函数可以用来在数组的指定位置插入一个或多个值。其语法如下：

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

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

下面我们通过一个示例来演示如何使用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]

在上面的示例中，我们将值10插入到索引为2的位置，原数组arr变为新数组new_arr。

在指定位置插入多个值

除了插入单个值，insert()函数还可以用来插入多个值。我们只需要将要插入的多个值放到一个列表中即可。例如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.insert(arr, 2, [10, 20, 30])

print(new_arr)

运行上面的代码，会输出：

[ 1  2 10 20 30  3  4  5]

在上面的示例中，我们在索引为2的位置插入了多个值。

在指定轴插入数据

如果我们的数组是多维数组，我们可以通过指定axis参数来在特定的轴上插入数据。例如：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = np.insert(arr, 1, [10, 20, 30], axis=0)

print(new_arr)

运行上面的代码，会输出：

[[ 1  2  3]
 [10 20 30]
 [ 4  5  6]]

在上面的示例中，我们在第一行的位置插入了新的一行。

总结

本文介绍了numpy中数组如何插入数据的方法，包括使用insert()函数在指定位置插入单个值或多个值，以及在多维数组中指定轴插入数据。通过灵活运用这些方法，我们可以方便地在numpy数组中插入新的数据。