python np.insert函数的用法

np.insert函数是NumPy库中的一个函数，用于在数组中插入元素。其语法如下：

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

其中，arr表示要插入元素的数组；obj表示要插入元素的位置；values表示要插入的元素；axis表示要插入元素的轴方向，默认为None，表示将数组展开后再插入元素。

示例：

import numpy as np

a = np.array([[1, 2], [3, 4], [5, 6]])
print('原始数组：\n', a)

# 在第一行插入元素
b = np.insert(a, 1, [7, 8], axis=0)
print('插入元素后的数组：\n', b)

# 在第二列插入元素
c = np.insert(a, 1, [7, 8, 9], axis=1)
print('插入元素后的数组：\n', c)

输出：

原始数组：
 [[1 2]
  [3 4]
  [5 6]]
插入元素后的数组：
 [[1 2]
  [7 8]
  [3 4]
  [5 6]]
插入元素后的数组：
 [[1 7 2]
  [3 8 4]
  [5 9 6]]