NumPy 数组操作 numpy.insert

这个函数沿着给定的轴在输入数组中的给定索引之前插入值。如果要插入的值的类型与输入数组不同，将进行类型转换。插入操作不会在原地进行，并且该函数会返回一个新数组。另外，如果未提及轴，则会对输入数组进行扁平化处理。

insert()函数接受以下参数 –

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

在这里，

序号	参数和描述
1	arr 输入数组
2	obj 要插入的位置之前的索引
3	values 要插入的值数组
4	axis 要插入的轴。如果未给出，则输入数组被展平

示例

import numpy as np 
a = np.array([[1,2],[3,4],[5,6]]) 

print 'First array:' 
print a 
print '\n'  

print 'Axis parameter not passed. The input array is flattened before insertion.'
print np.insert(a,3,[11,12]) 
print '\n'  
print 'Axis parameter passed. The values array is broadcast to match input array.'

print 'Broadcast along axis 0:' 
print np.insert(a,1,[11],axis = 0) 
print '\n'  

print 'Broadcast along axis 1:' 
print np.insert(a,1,11,axis = 1)

其输出将如下所示−

First array:
[[1 2]
 [3 4]
 [5 6]]

Axis parameter not passed. The input array is flattened before insertion.
[ 1 2 3 11 12 4 5 6]

Axis parameter passed. The values array is broadcast to match input array.
Broadcast along axis 0:
[[ 1 2]
 [11 11]
 [ 3 4]
 [ 5 6]]

Broadcast along axis 1:
[[ 1 11 2]
 [ 3 11 4]
 [ 5 11 6]]