【python】numpy使用

1、np.insert()：数组中插入元素。

# np.insert(arr, obj, values, axis)
# arr原始数组，可一可多，obj插入元素位置，values是插入内容，axis是按行按列插入。

插入的数组是一维的：

import numpy as np
a = np.array([1,4,6,5,6,8])
np.insert(a,0,9)
 
array([9, 1, 4, 6, 5, 6, 8])#插入元素都是在所给位置之前

插入的数组是多维的：

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

2、np.repeat()：对数组中的元素进行重复。

# np.repeat(a, repeats, axis=None)
# a是要重复的数组，repeats是指定每个元素重复的次数，而axis是指定重复的轴的参数。

import numpy as np
 
# 示例1：对一维数组进行元素重复
a = np.array([1, 2, 3])
b = np.repeat(a, repeats=3)
print(b)
# 输出：
# [1 1 1 2 2 2 3 3 3]
 
# 示例2：对二维数组按行进行元素重复
x = np.array([[1, 2], [3, 4]])
y = np.repeat(x, repeats=2, axis=0)
print(y)
# 输出：
# [[1 2]
#  [1 2]
#  [3 4]
#  [3 4]]
 
# 示例3：对二维数组按列进行元素重复
x = np.array([[1, 2], [3, 4]])
y = np.repeat(x, repeats=3, axis=1)
print(y)
# 输出：
# [[1 1 1 2 2 2]
#  [3 3 3 4 4 4]]

3、np.hstack()：在水平方向上（即沿着第二个轴）堆叠数组。

import numpy as np

# 创建两个二维数组，确保它们具有相同的行数
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 使用np.hstack()函数堆叠数组
c = np.hstack((a, b))

print(c)

# [[1 2 5 6]
# [3 4 7 8]]

4、numpy.unique() 函数接受一个数组，去除其中重复元素，并按元素由小到大返回一个新的无元素重复的元组或者列表。