numpy.insert()如何使用

這篇文章主要介紹“numpy.insert()如何使用”的相關(guān)知識(shí)，小編通過(guò)實(shí)際案例向大家展示操作過(guò)程，操作方法簡(jiǎn)單快捷，實(shí)用性強(qiáng)，希望這篇“numpy.insert()如何使用”文章能幫助大家解決問(wèn)題。

numpy.insert()主要用于向矩陣中插入行或列。對(duì)于多維矩陣，可以沿任意一個(gè)軸插入元素。

1. 參數(shù)說(shuō)明

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

arr：輸入矩陣，numpy.array類(lèi)型。注意：該方法并不改變直接arr的值，而是返回一個(gè)

obj：索引，整數(shù)或整數(shù)串。例如可以只插入一行元素，也可以插入多行元素，多行可以是連續(xù)的（如第0行和第1行），也可以是分立的（如第2行和第4行）。

values：插入的值，numpy.array類(lèi)型

axis：插入的軸，整型

返回值：得到的矩陣，numpy.array類(lèi)型

2. 示例

2.1. 插入一列，值為標(biāo)量

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

結(jié)果

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

2.2. 插入一列，值為一維矩陣

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

2.3. 插入多列，值為標(biāo)量

注意：索引值不能超過(guò)維度的最大索引，也就是說(shuō)只能在某一維的中間插值，不能在末尾插入。

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

結(jié)果

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

2.4. 輸入為一維向量

在向量 [1,2,3,4] 的第1個(gè)元素前面的位置插入5

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

在向量 [1,2,3,4] 的第1個(gè)元素前面的位置插入5，第2個(gè)元素前面的位置插入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:\n',np.insert(a,[1],[6,6],axis=0))

結(jié)果

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

關(guān)于“numpy.insert()如何使用”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)，可以關(guān)注億速云行業(yè)資訊頻道，小編每天都會(huì)為大家更新不同的知識(shí)點(diǎn)。