Python中数组元素的添加、插入和删除

数组

添加元素：使用np.append(x,[10,11])，作用为在末尾添加[10,11]
插入元素：使用np.insert(x,2,[10,11])，作用为在第2位置添加[10,11]
删除元素：使用np.delete(x,[1,2,3])，作用为删除第1,2,3位置的元素

Python中 numpy 数组元素的添加、插入和删除

一维数组创建：

添加元素：

使用np.append(x,[10,11])，作用为在末尾添加[10,11]

插入元素：

使用np.insert(x,3,[10,11])，作用为在第3位置添加[10,11]

删除元素：

使用np.delete(x,[1,2,3])，作用为删除第1,2,3位置的元素

本篇文章从append、 insert、 delete三个函数出发，讲解在numpy数组中如何添加和删除元素。

在 numpy中，经常需要对数组元素进行添加和删除操作，如使用 append() 函数和 insert() 函数为数组添加元素，或者使用 delete() 函数返回删除了某个轴的子数组的新数组，以及使用 unique() 函数寻找数组内的唯一元素。下面分别介绍这些函数的运用。

append()函数

该函数是在数组的末尾添加元素，该函数会返回一个新数组，而原数组不变。使用 append()函 数的方法如下：

 
<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'>numpy.append(arr,values,axis)<br></span></p>

 

对各个参数的说明：

参数 arr 表示输入的数组；

values 表示向 arr 数组添加的元素，values 为数组，values 数组列维度与 arr 数组列维度相同；

axis 表示沿着水平或竖直方向完成添加操作的轴，axis 取 0 表示沿竖直方向操作，axis 取 1 表示沿水平方向操作。

如果未提供 axis 值，在添加操作之前输入数组会被展开，values 可以是单元素，也可以是任意数组，将 values 添加到 arr 数组后，该函数会返回一个新数组，而原数组不变。

append() 函数的示例代码 example1 如下。

 
<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'># -*- coding: UTF-8 -*-

import numpy as np

#创建数组arr

arr = np.array([[1,2,3,4],[5,6,7,8]])

print('第1个数组arr：',arr)　　

print('向arr数组添加元素：')

print(np.append(arr,[[9,10],[11,12]]))

print('原数组：',arr)　　

print('沿轴 0 添加元素：')

print(np.append(arr,[[9,10,11,12],[11,11,11,11]], axis=0))

print('沿轴 1 添加元素：')

print(np.append(arr,[[9,10],[11,12]], axis=1))<br></span></p>

 

insert()函数

该函数在给定索引之前，沿给定轴在输入数组中插入值。该函数会返回一个新数组，原数组不变。使用 insert() 函数的方法如下：

<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'>numpy.insert(arr,obj,values,axis)<br></span></p>

 

对各个参数的说明：

参数 arr 表示输入的数组；

obj 表示在其之前插入值的索引；

values 表示向arr数组插入的值，values值可为单元素或values数组，并且 values 数组行维度与arr数组列维度相同；

axis 表示沿着水平或竖直方向完成插入操作的轴，axis 取 0 表示沿竖直方向操作，即在 arr 数组行索引位于 obj 的位置处插入 values 值，axis 取 1 表示沿水平方向操作，即在 arr 数组列索引位于 obj 的位置处插入 values 值。

如果未提供 axis 值，在插入之前输入数组会被展开，values 可以是单个元素，也可以是一维数组，将 values 插入到 obj 的位置处，该函数会返回一个新数组，原数组不变。

另外，insert() 函数如果传递了 axis 参数，则插入 values 值时，会以广播值数组来配输入数组，即 np.insert(arr,2,[9],axis=0)，其中 values 为 [9]，arr 是 2 行 4 列数组，由于 axis=0，则插入值的数组列维度与arr数组列维度相同，因此，插入值数组为 1 行 4 列的数组，也就是广播值数组 [9,9,9,9]。

insert() 函数的示例代码 example2 如下。

 
<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'># -*- coding: UTF-8 -*-

import numpy as np

#创建数组arr

arr = np.array([[1,2,3,4], [5,6,7,8]])

print('第1个数组arr：',arr)　　

print('未传递axis参数。在插入之前输入数组会被展开。')

print(np.insert(arr,4,[9, 10,11,12]))

print('传递了axis参数,会以广播值数组来配输入数组。')

print('沿轴 0 广播：')

print(np.insert(arr, 2, [9], axis=0))    #插入值是[9,9,9,9]

print('沿轴 0 插入：')

print(np.insert(arr, 2, [9,10,11,12], axis=0))

print('沿轴 1 广播：')

print(np.insert(arr, 2, [[9],[10]], axis=1))

print('沿轴 1 插入：')

print(np.insert(arr, 2, [[9,10],[11,12]], axis=1))<br></span></p>

该函数返回从输入数组中删除指定子数组的新数组，原数组不变。与 insert() 函数的情况一样，如果未提供轴参数，则输入数组将展开。使用 delete() 函数的方法如下：

<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'>numpy.delete(arr,obj,axis)<br></span></p>

对各个参数的说明：

参数 arr 表示输入的数组；

obj 是用整数或者整数数组表示的从输入数组中删除的子数组，obj 可以用切片 numpy.s_[start🔚step] 表示要删除的子数组范围；

axis 表示沿着它删除给定子数组的轴，如果未提供 axis 值，则输入数组会被展开。

delete() 函数的示例代码 example3 如下。

 
<p style="line-height: 1.75em;"><span style='font-family: 微软雅黑, "Microsoft YaHei";'># -*- coding: UTF-8 -*-

import numpy as np

#创建数组arr

arr = np.arange(16).reshape(4,4)

print('第1个数组arr：',arr)

print ('未传递 axis 参数，在删除之前输入数组会被展开。')

print (np.delete(arr,[6,9,12]))   #删除arr数组中索引为6，9，12的元素

print(arr)

print('删除第3列：')

print(np.delete(arr, 2, axis=1))

print('删除第2行：')

print(np.delete(arr, 1, axis=0))　　

print('从数组中删除用切片表示元素范围值：')

print(np.delete(arr, np.s_[::2]))<br></span></p>

 

以上就是三种函数在numpy数组中的运用，更多Python学习推荐:PyThon学习网教学中心。