python可视化操作(一)读取excel数据绘制条形图(直方图)

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属性

你可能感兴趣的:(Python)