Numpy API接口说明

持续更新…

np.hstack()

功能：用于多个数组沿着水平轴（即列）连接起来，即沿着水平轴（即列）将输入的数组堆叠起来。输出数组的列数是输入数组的列数之和，行数是输入数组中最大行数的值。

import numpy as np
 
arr1 = np.array([[1,2], [3,4] ])
arr2 = np.array([[5,6], [7,8] ])
res = np.hstack((arr1, arr2))
print (res)

输出：

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

输出：

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.hstack((a, b))
print(c)  

[1 2 3 4 5 6]

np.vstack()

功能：用于多个数组沿着垂直轴（即行）连接起来，即沿着垂直轴（即行）将输入的数组堆叠起来。输出数组的行数是输入数组的行数之和，列数是输入数组中最大列数的值。

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
c = np.vstack((a, b))
print(c)

输出：

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

np.random.choice()

np.random.choice(a, size=None, replace=True, p=None)
功能：随机抽样函数，用于从给定的一维数组中随机抽取指定数量或指定概率的元素。
参数：

1. a：一维数组或整数，表示需要进行抽取的数据源，当为整数时，相当于 np.arange(n)。
2. size：整数或元组，表示输出数组的大小。
3. replace：布尔值，表示是否允许重复抽样，默认为 True（允许重复抽样）。
4. p：一维数组，表示每个元素被抽到的概率。如果未指定，则默认为均匀分布。

import numpy as np
a = np.array([1, 2, 3, 4])
# 随机抽取1个元素
print(np.random.choice(a))   
# 随机抽取2个元素，2个元素可相同
print(np.random.choice(a, size=2))   
# 随机抽取2个元素，2个元素不相同，即无放回抽取
print(np.random.choice(a, size=2, replace=False))
# 按照概率分布随机抽取3个元素
print(np.random.choice(a, size=3, p=[0.1, 0.2, 0.2, 0.3]))