python的numpy中有哪些常用的的函数
# Python的NumPy中有哪些常用的函数
## 目录
1. [NumPy简介](#numpy简介)
2. [数组创建函数](#数组创建函数)
3. [数组操作函数](#数组操作函数)
4. [数学运算函数](#数学运算函数)
5. [统计函数](#统计函数)
6. [线性代数函数](#线性代数函数)
7. [随机数生成函数](#随机数生成函数)
8. [文件输入输出函数](#文件输入输出函数)
9. [其他实用函数](#其他实用函数)
10. [总结](#总结)
## NumPy简介
NumPy(Numerical Python)是Python科学计算的基础库,提供了高性能的多维数组对象和用于操作这些数组的工具。它是数据科学、机器学习、科学计算等领域的核心库之一。
NumPy的主要特点包括:
- 强大的N维数组对象ndarray
- 广播功能函数
- 整合C/C++/Fortran代码的工具
- 线性代数、傅里叶变换、随机数生成等功能
```python
import numpy as np # 常规导入方式
数组创建函数
1. 基础创建函数
| 函数 |
描述 |
示例 |
np.array() |
从列表或元组创建数组 |
np.array([1,2,3]) |
np.zeros() |
创建全零数组 |
np.zeros((3,3)) |
np.ones() |
创建全1数组 |
np.ones((2,2)) |
np.empty() |
创建未初始化数组 |
np.empty((3,2)) |
np.arange() |
类似range的数组 |
np.arange(0,10,2) |
np.linspace() |
等间隔数组 |
np.linspace(0,1,5) |
np.eye() |
单位矩阵 |
np.eye(3) |
np.diag() |
对角矩阵 |
np.diag([1,2,3]) |
2. 特殊数组创建
# 从已有数据创建
np.copy(a) # 数组副本
np.fromfunction(func, shape) # 通过函数创建
# 网格生成
np.mgrid[0:5,0:5] # 密集网格
np.ogrid[0:5,0:5] # 开放网格
数组操作函数
1. 形状操作
| 函数 |
描述 |
示例 |
reshape() |
改变数组形状 |
arr.reshape(3,4) |
resize() |
改变数组形状和大小 |
np.resize(arr, (5,5)) |
flatten() |
展平为一维数组 |
arr.flatten() |
ravel() |
返回视图的展平数组 |
arr.ravel() |
transpose() |
转置数组 |
arr.transpose() |
swapaxes() |
交换两个轴 |
arr.swapaxes(0,1) |
2. 数组连接与分割
# 连接
np.concatenate((a1,a2), axis=0)
np.vstack((a1,a2)) # 垂直堆叠
np.hstack((a1,a2)) # 水平堆叠
np.column_stack((a1,a2)) # 按列堆叠
# 分割
np.split(arr, indices_or_sections)
np.vsplit(arr, 3) # 垂直分割
np.hsplit(arr, 2) # 水平分割
3. 数组增删
np.append(arr, values, axis=None)
np.insert(arr, obj, values, axis=None)
np.delete(arr, obj, axis=None)
np.unique(arr) # 去重
数学运算函数
1. 基础数学运算
| 函数 |
描述 |
示例 |
add() |
加法 |
np.add(a,b) |
subtract() |
减法 |
np.subtract(a,b) |
multiply() |
乘法 |
np.multiply(a,b) |
divide() |
除法 |
np.divide(a,b) |
power() |
幂运算 |
np.power(a,2) |
mod() |
取模 |
np.mod(a,3) |
abs() |
绝对值 |
np.abs(a) |
2. 三角函数
np.sin(arr) # 正弦
np.cos(arr) # 余弦
np.tan(arr) # 正切
np.arcsin(arr) # 反正弦
np.arccos(arr) # 反余弦
np.arctan(arr) # 反正切
np.hypot(x1, x2) # 欧几里得范数
3. 指数对数函数
np.exp(arr) # e的幂次
np.expm1(arr) # exp(x)-1
np.log(arr) # 自然对数
np.log10(arr) # 10为底对数
np.log2(arr) # 2为底对数
np.log1p(arr) # log(1+x)
4. 舍入函数
np.around(arr, decimals=0) # 四舍五入
np.floor(arr) # 向下取整
np.ceil(arr) # 向上取整
np.trunc(arr) # 截断小数部分
np.rint(arr) # 四舍五入到整数
统计函数
1. 基础统计
| 函数 |
描述 |
示例 |
amin() |
沿轴最小值 |
np.amin(arr, axis=0) |
amax() |
沿轴最大值 |
np.amax(arr, axis=1) |
ptp() |
范围(最大值-最小值) |
np.ptp(arr) |
percentile() |
百分位数 |
np.percentile(arr, 75) |
median() |
中位数 |
np.median(arr) |
mean() |
算术平均数 |
np.mean(arr) |
average() |
加权平均数 |
np.average(arr, weights=w) |
2. 方差与标准差
np.var(arr) # 方差
np.std(arr) # 标准差
np.cov(m, y=None) # 协方差矩阵
np.corrcoef(x, y=None) # 相关系数
3. 直方图统计
np.histogram(a, bins=10) # 直方图
np.bincount(x, weights=None) # 整型数组的计数
np.digitize(x, bins) # 返回每个值所属的bin索引
线性代数函数
1. 矩阵操作
| 函数 |
描述 |
示例 |
dot() |
点积 |
np.dot(a,b) |
vdot() |
向量点积 |
np.vdot(a,b) |
inner() |
内积 |
np.inner(a,b) |
outer() |
外积 |
np.outer(a,b) |
matmul() |
矩阵乘法 |
np.matmul(a,b) |
tensordot() |
张量点积 |
np.tensordot(a,b) |
2. 矩阵分解与求逆
np.linalg.inv(a) # 矩阵求逆
np.linalg.pinv(a) # 伪逆矩阵
np.linalg.det(a) # 行列式
np.linalg.eig(a) # 特征值和特征向量
np.linalg.svd(a) # 奇异值分解
np.linalg.qr(a) # QR分解
np.linalg.cholesky(a) # Cholesky分解
3. 线性方程组
np.linalg.solve(a, b) # 解线性方程组
np.linalg.lstsq(a, b) # 最小二乘解
np.linalg.norm(x) # 范数计算
随机数生成函数
1. 基础随机数
| 函数 |
描述 |
示例 |
rand() |
[0,1)均匀分布 |
np.random.rand(3,2) |
randn() |
标准正态分布 |
np.random.randn(100) |
randint() |
随机整数 |
np.random.randint(0,10,size=5) |
random() |
[0,1)随机浮点数 |
np.random.random(10) |
choice() |
随机选择 |
np.random.choice(a, size=3) |
shuffle() |
随机打乱 |
np.random.shuffle(arr) |
permutation() |
随机排列 |
np.random.permutation(10) |
2. 概率分布
# 连续分布
np.random.uniform(low=0.0, high=1.0) # 均匀分布
np.random.normal(loc=0.0, scale=1.0) # 正态分布
np.random.exponential(scale=1.0) # 指数分布
# 离散分布
np.random.binomial(n=10, p=0.5) # 二项分布
np.random.poisson(lam=1.0) # 泊松分布
3. 随机种子
np.random.seed(42) # 设置随机种子
np.random.get_state() # 获取随机状态
np.random.set_state(state) # 设置随机状态
文件输入输出函数
1. 文本文件
| 函数 |
描述 |
示例 |
loadtxt() |
从文本加载 |
np.loadtxt('data.txt') |
savetxt() |
保存到文本 |
np.savetxt('out.txt', arr) |
genfromtxt() |
从文本生成数组 |
np.genfromtxt('data.csv', delimiter=',') |
2. 二进制文件
np.save('array.npy', arr) # 保存为.npy
np.load('array.npy') # 加载.npy文件
np.savez('arrays.npz', a=arr1, b=arr2) # 保存多个数组
np.savez_compressed('arrays.npz', a=arr1) # 压缩保存
3. 内存映射
np.memmap(filename, dtype='float32', mode='r', shape=(100,100))
其他实用函数
1. 索引与切片
np.where(condition, [x, y]) # 条件选择
np.take(a, indices) # 按索引获取元素
np.select(condlist, choicelist) # 多条件选择
np.nonzero(a) # 非零元素索引
2. 逻辑运算
np.logical_and(x1, x2)
np.logical_or(x1, x2)
np.logical_not(x)
np.logical_xor(x1, x2)
np.all(a) # 所有元素为True
np.any(a) # 任一元素为True
3. 数组比较
np.equal(a, b) # ==
np.not_equal(a, b) # !=
np.greater(a, b) # >
np.greater_equal(a, b) # >=
np.less(a, b) # <
np.less_equal(a, b) # <=
4. 集合操作
np.intersect1d(ar1, ar2) # 交集
np.union1d(ar1, ar2) # 并集
np.setdiff1d(ar1, ar2) # 差集
np.setxor1d(ar1, ar2) # 对称差集
np.in1d(ar1, ar2) # 测试成员
5. 广播与矢量化
np.broadcast(arr1, arr2) # 广播对象
np.vectorize(pyfunc) # 矢量化函数
总结
NumPy提供了丰富的函数库,涵盖了从数组创建、数学运算到线性代数和随机数生成的各个方面。掌握这些常用函数能够显著提高科学计算和数据分析的效率。本文介绍了约150个常用NumPy函数,但NumPy的功能远不止于此。建议读者在实际应用中结合官方文档深入探索更多高级功能。
学习建议
- 从基础数组操作开始练习
- 熟悉广播机制和矢量化运算
- 掌握线性代数和统计相关函数
- 在实际项目中应用这些函数
- 查阅官方文档获取最新功能
NumPy作为Python科学计算生态系统的基石,其重要性不言而喻。希望本文能帮助您更好地理解和使用NumPy的强大功能。
# 示例:综合应用
arr = np.random.randn(100, 100) # 创建100x100正态分布数组
arr = arr * 10 + 5 # 线性变换
mean = np.mean(arr) # 计算均值
std = np.std(arr) # 计算标准差
normalized = (arr - mean) / std # 标准化
eigenvalues = np.linalg.eig(normalized) # 特征值分解
注意:本文基于NumPy 1.21+版本,部分函数在早期版本中可能不可用或行为不同。
“`