1 # -*- coding: utf-8 -*- 2 import numpy as np 3 import matplotlib.pyplot as plt 4 import pandas as pd 5 import math 6 plt.rcParams['font.sans-serif'] = ['SimHei'] 7 plt.rcParams['axes.unicode_minus'] = False 8 # matplotlib画图中中文显示会有问题，需要这两行设置默认字体 9 10 plt.xlabel('n') 11 plt.ylabel('Time/ns') 12 plt.xlim(xmax=30, xmin=5) 13 plt.ylim(ymax=50, ymin=5) 14 # 画两条（0-9）的坐标轴并设置轴标签x，y 15 data1 = [8.28, 10.76, 12.88, 13.2, 15.08, 15.6, 17.64, 19.0, 21.08, 20.82, 16 22.52, 24.8, 23.6, 27.86, 29.72,32.38, 32.14, 37.68, 35.6, 36.44, 17 35.96, 38.6, 39.52, 40.7, 42.74, 44.52] 18 data2 = [9.7, 11.8, 13.48, 15.78, 15.36, 17.84, 18.64, 19.76, 21.96, 22.42, 19 24.48, 31.08, 29.8, 31.26, 32.8, 36.24, 35.46, 39.0, 35.58, 41.64, 20 41.42, 42.44, 44.46, 45.22, 46.06, 47.44] 21 plt.title("约瑟夫问题的两种解法时间分析") 22 23 x1 = np.arange(5,31,1) 24 y1 = np.array(data1) 25 y2 = np.array(data2) 26 #线性拟合 27 p1 = np.polyfit(x1,y1,1) 28 ry1 = np.polyval(p1,x1) 29 p2 = np.polyfit(x1,y2,1) 30 ry2 = np.polyval(p2,x1) 31 32 colors1 = '#fab1a0' # 点的颜色 33 colors2 = '#0984e3' 34 area = np.pi * 4 ** 2 # 点面积 35 # 画散点图 36 plt.plot(x1,ry1,colors1,label='数组_拟合曲线') 37 plt.plot(x1,ry2,colors2,label ='链表_拟合曲线') 38 plt.scatter(x1, y1, s=area, c=colors1, alpha=0.8, label='循环数组') 39 plt.scatter(x1, y2, s=area, c=colors2, alpha=0.8, label='循环链表') 40 #plt.plot([5, 30], [8.9, 46], linewidth='1', color='#000000') 41 plt.legend() 42 #plt.grid('True') 43 plt.savefig('1.png', dpi=300) 44 plt.show() 45 ''' 46 df = pd.DataFrame(columns=['人数n'],data = x1) 47 df['CArray_time'] = data1 48 df['CList_time'] = data2 49 print(df) 50 print('') 51 print(df.corr('pearson')) 52 '''