numpy.insert()如何使用

栏目分类：其他教程    发布日期：2023-09-01    浏览次数：258次     收藏

这篇文章主要介绍“numpy.insert()如何使用”的相关知识，小编通过实际案例向大家展示操作过程，操作方法简单快捷，实用性强，希望这篇“numpy.insert()如何使用”文章能帮助大家解决问题。

numpy.insert()主要用于向矩阵中插入行或列。对于多维矩阵，可以沿任意一个轴插入元素。

1. 参数说明

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

arr：输入矩阵，numpy.array类型。注意：该方法并不改变直接arr的值，而是返回一个

obj：索引，整数或整数串。例如可以只插入一行元素，也可以插入多行元素，多行可以是连续的（如第0行和第1行），也可以是分立的（如第2行和第4行）。

values：插入的值，numpy.array类型

axis：插入的轴，整型

返回值：得到的矩阵，numpy.array类型

2. 示例

2.1. 插入一列，值为标量

a = np.array([[1, 1], [2, 2], [3, 3]])
np.insert(a, 1, 5, axis=1)

结果

array([[1, 5, 1],
[2, 5, 2],
[3, 5, 3]])

2.2. 插入一列，值为一维矩阵

np.insert(a, [1], [[1],[2],[3]], axis=1)

2.3. 插入多列，值为标量

注意：索引值不能超过维度的最大索引，也就是说只能在某一维的中间插值，不能在末尾插入。

x = np.arange(8).reshape(2, 4)
idx = (1, 3)
np.insert(x, idx, 999, axis=1)

结果

array([[ 0, 999, 1, 2, 999, 3],
[ 4, 999, 5, 6, 999, 7]])

2.4. 输入为一维向量

在向量 [1,2,3,4] 的第1个元素前面的位置插入5

print(np.insert([1,2,3,4],1,5))
[1 5 2 3 4]

在向量 [1,2,3,4] 的第1个元素前面的位置插入5，第2个元素前面的位置插入7

print(np.insert([1,2,3,4],[1,2],[5,7]))
[1 5 2 7 3 4]

2.5. 输入为矩阵

插入一整行

import numpy as np
a = a = np.array([[1, 1], [2, 2], [3, 3]])
print('a=',a)
print('after insertion:
',np.insert(a,[1],[6,6],axis=0))

结果

a= [[1 1]
[2 2]
[3 3]]
after insertion:
[[1 1]
[6 6]
[2 2]
[3 3]]