python代码根据数据画图

1、根据两组数据画折线图

代码如下

​
# pyplot 是专门用来画图的存在于matplotlib里面的函数
from matplotlib import pyplot as plt

x1 = [1000,3000,5000,7000,9000]


TPR_y1 = [0,0.48,0.93,0.95,0.95]

FPR_y2 = [0,0.81,0.07,0.04,0.01]


plt.xlim([1000, 10000])
plt.ylim([0.0, 1.0]) 


# 绘图
plt.plot(x1, TPR_y1,color='red',label = 'TPR')  # 传入x，y绘制出折线图

plt.plot(x1, FPR_y2,color='blue',label = 'FPR')

plt.legend()

plt.savefig("./sig_size.png")  # 保存图片 要在绘制之后保存 plt.savefig保存图片 保存为svg这种矢量图，放大不会有锯齿
plt.xlabel('times')
plt.ylabel('TPR/FPR')
# 展示图形
plt.show()  # 在执行程序的同时显示出图形

​

2、根据数据画柱状图

import numpy as np
import matplotlib.pyplot as plt
import random



# 准备数据
x_data = [f"20{i}年" for i in range(16, 21)]
y_data = [random.randint(100, 300) for i in range(6)]

# 正确显示中文和负号
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False

# 画图，plt.bar()可以画柱状图
for i in range(len(x_data)):
	plt.bar(x_data[i], y_data[i])
# 设置图片名称
plt.title("销量分析")
# 设置x轴标签名
plt.xlabel("年份")
# 设置y轴标签名
plt.ylabel("销量")
# 显示
plt.show()