Python 数据处理之折线图--matplotlib

# -*- coding:utf-8 -*-
import matplotlib.pyplot as plt
import xlrd
import os


workbook = xlrd.open_workbook(os.path.join(os.getcwd(), 'Summary.xlsx'))
sheetData = workbook.sheets()[0]

rowNum = sheetData.nrows   #获取行数
colNum = sheetData.ncols   #获取列数

Ver = sheetData.row_values(0, 2, colNum)  #获取第0行，第2-colNum列(列从0列开始)
data1 = sheetData.row_values(3, 2, colNum)
data2 = sheetData.row_values(4, 2, colNum)
data3 = sheetData.row_values(5, 2, colNum)
data4 = sheetData.row_values(6, 2, colNum)

plt.figure()  # 创建画布
plt.plot(Ver, data1, marker='.', color='r', label='PCMark-data1')
plt.plot(Ver, data2, marker='.', color='b', label='PCMark-data2')
plt.plot(Ver, data3, marker='.', color='g', label='PCMark-data3')
plt.plot(Ver, data4, marker='.', color='c', label='PCMark-data4')
plt.legend()  # 显示图例（使绘制生效）
plt.xlabel('DeviceVer')  # 横坐标名称
plt.ylabel('Score')  # 纵坐标名称
plt.title('PCMark10 Core')  # 表格名称
plt.savefig(os.path.join(os.getcwd(), 'Summary'))  # 保存图片Summary.png到本地
plt.show()  # 显示图片