Numpy 数组元素添加与元素删除函数详解

元素添加

Numpy中有类似python列表操作函数append()及insert()，但是用法稍有不同，append()及insert()不作为数组的实例方法使用。

np.append()

np.append()的参数如下

def append(arr, values, axis=None):

其中，arr为数组对象，values为要插入的值或子数组，axis为0则沿行方向添加元素，为1则沿列方向添加元素，但都是将元素加至末尾

插入一维数组

import numpy as npdata1 = np.random.randint(0,10,5)
print(data1)data1 = np.append(data1, 100)print(data1)
# 
# [1 0 3 1 7]
# [  1   0   3   1   7 100]

插入二维数组

插入二维数组就会涉及到插入的方式，也就是沿行添加还是沿列添加，并且要保证数组每个维度的长度相等，而且如果指定插入的轴，则会将数组扁平化处理，也就是降维至一维数组，并且插入的数组维度要与原数组相同

不指定插入的轴

import numpy as npdata1 = np.random.randint(0,10,(3,4))
print(data1)
data2 = np.append(data1,[[1,1,1,1]])
print(data2)# [[6 1 3 4]
#  [7 6 2 6]
#  [6 5 7 4]]
# [6 1 3 4 7 6 2 6 6 5 7 4 1 1 1 1]

可以看到数组做了扁平化处理，然后将元素添加到了末尾的位置 

沿行插入

import numpy as npdata1 = np.random.randint(0,10,(3,4))
print(data1)
data2 = np.append(data1,[[1,1,1,1]], axis=0)
print(data2)# [[9 2 7 3]
#  [6 4 1 8]
#  [6 2 1 1]]
# [[9 2 7 3]
#  [6 4 1 8]
#  [6 2 1 1]
#  [1 1 1 1]]

沿列插入

import numpy as npdata1 = np.random.randint(0,10,(3,4))
print(data1)
data2 = np.append(data1,np.ones((3,1)), axis=1)
print(data2)# [[9 4 1 6]
#  [6 7 1 0]
#  [0 2 5 6]]
# [[9. 4. 1. 6. 1.]
#  [6. 7. 1. 0. 1.]
#  [0. 2. 5. 6. 1.]]

 注意这里的维度要保持一致，也就是形状要为（3，1）

insert()

insert()函数的作用为指定索引位置插入元素或子数组，参数如下

def insert(arr, obj, values, axis=None):

可以看到arr为数组对象，obj为索引位置，values为要插入的元素或子数组，axis为插入的轴

同样这里要保准数组维度，并且不指定axis会导致数组扁平化

插入一维数组

import numpy as npdata1 = np.random.randint(0,10,5)
print(data1)data2 = np.insert(data1,2,100)
print(data2)
# [8 0 6 5 1]
# [  8   0 100   6   5   1]

插入二维数组

不指定插入的轴

import numpy as npdata1 = np.random.randint(0,10,(2,3))
print(data1)data2 = np.insert(data1,2,[1,2,3])
# [[7 3 9]
#  [3 1 3]]
# [7 3 1 2 3 9 3 1 3]

可以看到这里直接将数组扁平化后，将要插入的子数组按元素插入了指定的索引位置

沿行插入

import numpy as npdata1 = np.random.randint(0,10,(2,3))
print(data1)data2 = np.insert(data1,2,[[1,2,3]],axis=0)
print(data2)
# [[1 3 3]
#  [2 3 0]]
# [[1 3 3]
#  [2 3 0]
#  [1 2 3]]

沿列插入

import numpy as npdata1 = np.random.randint(0,10,(2,3))
print(data1)data2 = np.insert(data1,2,np.ones((2,1)),axis=1)
print(data2)
# [[5 9 5]
#  [6 0 3]]
# [[5 9 1 1 5]
#  [6 0 1 1 3]]

补充：insert()函数的广播机制

insert函数在进行二维数组的插入时，不同于append()函数需要插入数组的维度要与被插入的数组一致，insert()函数可以将插入数组进行广播，从而匹配形状

import numpy as npdata1 = np.random.randint(0,10,(2,3))
print(data1)data2 = np.insert(data1,2,1,axis=1)
data3 = np.insert(data1,2,[1,1,1],axis=0)
print(data2)
print(data3)
# [[2 8 7]
#  [5 8 9]]
# [[2 8 1 7]
#  [5 8 1 9]]
# [[2 8 7]
#  [5 8 9]
#  [1 1 1]]

这里分别使用常量1和一维数组作为插入对象进行了广播尝试

元素删除

np.delete()

元素删除这里使用np.delete()函数，函数参数如下

def delete(arr, obj, axis=None):

arr为要进行删除的数组对象，obj为要删除的元素对应的下标，axis为要进行删除的轴

同样，若不指定axis，则会对数组进行扁平化处理

对一维数组进行删除

import numpy as npdata1 = np.random.randint(0,10,5)
print(data1)data2 = np.delete(data1,2)
print(data2)
# 
# [1 1 4 3 6]
# [1 1 3 6]

对二维数组进行阐述

不指定轴

import numpy as npdata1 = np.random.randint(0,10,(2,3))
print(data1)data2 = np.delete(data1,2)
print(data2)
# [[5 4 6]
#  [0 3 7]]
# [5 4 0 3 7]

可以看到数组扁平化后删除了索引为2的值

删除指定行

import numpy as npdata1 = np.random.randint(0,10,(3,4))
print(data1)data2 = np.delete(data1,2,axis=0)
print(data2)
# [[8 5 7 2]
#  [9 0 7 1]
#  [6 2 9 0]]
# [[8 5 7 2]
#  [9 0 7 1]]

删除指定列

import numpy as npdata1 = np.random.randint(0,10,(3,4))
print(data1)data2 = np.delete(data1,2,axis=1)
print(data2)
# [[3 5 0 2]
#  [0 0 0 2]
#  [4 2 8 6]]
# [[3 5 2]
#  [0 0 2]
#  [4 2 6]]