import xlrd
import matplotlib.pyplot as plt
import numpy as np
import os
def read_excel():
print(os.getcwd())
#打开excel文件
workbook = xlrd.open_workbook('data.xlsx')
#workbook = xlrd.open_workbook(os.path.join(os.getcwd(), 'data.xlsx'))
#打开sheet1
sheet = workbook.sheet_by_name("Sheet1")
#获取行数
row_num = sheet.nrows
row_list = []
for i in range(1, row_num):
#获得第i行的数据
row_data = sheet.row_values(i)
for data in row_data:
if data != '':
row_list.append(data)
# for row in row_list:
# print (row)
return sorted(row_list)
def draw_picture(row_list):
plt.ylabel("speed of Pedestrians 行人速度", fontproperties='SimHei')
plt.bar(range(len(row_list)), row_list)
#平均数
print("平均数:", np.mean(row_list))
print("平均数:" + str(np.mean(row_list)))
#中位数
print("中位数:", np.median(row_list))
plt.show()
if __name__ == '__main__':
draw_picture(read_excel())
在这里一共四个类库,其用途分别是:
- xlrd操作excel
- matplotlib.pyplot绘图
- numpy算平均数中位数
- os寻找当前路径
中间遇到的一些问题:
我在用VSCode中执行workbook = xlrd.open_workbook(‘data.xlsx’)的时候会提示Python/Excel - IOError: [Errno 2] No such file or directory。这点让我觉得很费解,因为一般来说,程序执行的时候如果打开其他文件可以直接写上文件名(也就是用上相对位置),但是这里却出了问题。
经过观察(打印os.getcwd()),发现是因为VSCode的命令行默认的执行路径不是目标文件的路径,而是与cmd一样默认的C:\Users…>。
在pycharm中执行的时候不会出现这个错误。为了让程序的鲁棒性更好,我把语句改成了workbook = xlrd.open_workbook(os.path.join(os.getcwd(), ‘data.xlsx’))在使用matplotlib.pyplot绘制图形的时候出现中文乱码。参考网上的做法, plt.ylabel(“speed of Pedestrians 行人速度”, fontproperties=’SimHei’),加上了fontproperties属性