Python中numpy数组如何添加和删除元素

发布网友发布时间：2024-08-30 13:44

共1个回答

热心网友时间：2024-08-30 13:58

在Python的numpy库中，对数组元素进行增删操作相当直接。主要借助append(), insert(), 和delete()这三个函数。让我们一一解析它们的用法和示例。

首先，append()函数用于在数组尾部添加元素，返回的是新数组，原数组保持不变。其用法如下：
如果axis未指定，append会展平数组，然后将values（可以是单一元素或数组）添加到arr，返回新数组。例如：

python
arr = np.array([1, 2, 3])
new_arr = np.append(arr, 4)

接下来是insert()函数，它会在指定索引前插入值，同样返回新数组。参数说明包括：
- 不指定axis时，数组会被展平。
- values可以是单值或一维数组。
- 如果提供了axis，values会被广播到与输入数组相同的维度。例如：
python
arr = np.array([[1, 2], [3, 4]])
new_arr = np.insert(arr, 1, [5], axis=1)

最后，delete()函数用于移除指定的子数组，返回删除后的数组。如果axis未指定，数组会被展平进行操作。示例如下：
python
arr = np.array([1, 2, 3, 4])
new_arr = np.delete(arr, slice(1, 3))

通过这些函数，你可以灵活地在numpy数组中进行元素的添加和删除操作，保持或改变数组结构。