Python绘图库Matplotlib:展示算法运行时间
2024.02.16 18:08浏览量:8简介:使用Python的Matplotlib库绘制算法运行时间图表,帮助理解算法的效率。本文将介绍如何使用Matplotlib来可视化算法的运行时间,并通过图表展示不同数据规模下算法的性能表现。
首先,我们需要安装matplotlib库。在命令行中输入以下命令进行安装:
pip install matplotlib
接下来,我们将通过一个简单的例子展示如何使用matplotlib来绘制算法运行时间图表。在这个例子中,我们将模拟一个简单的排序算法,并记录不同数据规模下的运行时间。
```python
import matplotlib.pyplot as plt
import time
定义排序函数
def sort_algorithm(arr):
# 这里我们简单地使用冒泡排序作为示例for i in range(len(arr)):for j in range(len(arr) - i - 1):if arr[j] > arr[j + 1]:arr[j], arr[j + 1] = arr[j + 1], arr[j]return arr
生成随机数据
data_sizes = [100, 200, 500, 1000, 2000] # 不同的数据规模
times = [] # 存储运行时间
记录每个数据规模下的运行时间
for data_size in data_sizes:
arr = list(range(data_size)) # 生成随机整数数组
start_time = time.time() # 记录开始时间
sort_algorithm(arr) # 调用排序函数
end_time = time.time() # 记录结束时间
times.append(end_time - start_time) # 计算运行时间并存储
使用matplotlib绘制运行时间图表
plt.figure(figsize=(10, 6)) # 设置图表大小
plt.plot(data_sizes, times, marker=’o’) # 绘制运行时间曲线
plt.xlabel(‘Data size’) # 设置x轴标签
plt.ylabel(‘Running time (seconds)’) # 设置y轴标签
plt.title(‘Running time of sorting algorithm’) # 设置图表标题
plt.show() # 显示图表

发表评论
登录后可评论,请前往 登录 或 注册