logo

Python绘图库Matplotlib:展示算法运行时间

作者:4042024.02.16 18:08浏览量:8

简介:使用Python的Matplotlib库绘制算法运行时间图表,帮助理解算法的效率。本文将介绍如何使用Matplotlib来可视化算法的运行时间,并通过图表展示不同数据规模下算法的性能表现。

首先,我们需要安装matplotlib库。在命令行中输入以下命令进行安装:

  1. pip install matplotlib

接下来,我们将通过一个简单的例子展示如何使用matplotlib来绘制算法运行时间图表。在这个例子中,我们将模拟一个简单的排序算法,并记录不同数据规模下的运行时间。

```python
import matplotlib.pyplot as plt
import time

定义排序函数

def sort_algorithm(arr):

  1. # 这里我们简单地使用冒泡排序作为示例
  2. for i in range(len(arr)):
  3. for j in range(len(arr) - i - 1):
  4. if arr[j] > arr[j + 1]:
  5. arr[j], arr[j + 1] = arr[j + 1], arr[j]
  6. 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() # 显示图表

相关文章推荐

发表评论