标签：plt name Python 查询 评分 print input data

这是接的小单，具体内容如下

通过python对数据csv的读取，完成数据添加，显示，查询和统计。

下图是该csv 数据。

Python修改代码实现添加，显示，查询和统计电影评分数据

实现的代码如下所示。

import pandas as pd 
import numpy as np
import matplotlib as mpl
from xlrd import open_workbook
from xlutils.copy import copy
from matplotlib import pyplot as plt
mpl.rcParams['font.sans-serif'] = ['SimHei']
mpl.rcParams['axes.unicode_minus'] = False


def add():# 添加数据函数
    # 打开文件，写入数据
    # 接收用户收入
    id = input('请输入电影号')
    name = input('请输入名称')
    fen  = input('请输入评分')
    # 向列表中添加数据
    data.append([id, name,fen])

    writer = pd.ExcelWriter('253.xlsx',)
    r_xls = open_workbook("253.xlsx") # 读取excel文件
    row = r_xls.sheets()[0].nrows # 获取已有的行数
    excel = copy(r_xls) # 将xlrd的对象转化为xlwt的对象
    table = excel.get_sheet(0) # 获取要操作的sheet

    #对excel表追加一行内容
    table.write(row, 0, id) #括号内分别为行数、列数、内容
    table.write(row, 1, name)
    table.write(row, 2, fen)

    excel.save("253.xlsx") # 保存并覆盖文件
    print('添加成功')


def show():  # 显示函数
    # 打印标题
    print('编号 名称 评分')
    # 遍历数据列表 然后打印数据
    for i in data:
        for j in i:
            print(j, end=' ')
        print()


def quety():  # 查询函数
    # 选择查询目标
    flag = int(input('1.按照编号查询  2.按照名称查询'))
    if flag == 1:
        id = input('输入编号')
        # 职工编号列表
        ids = [str(i[0]) for i in data]
        # 判断输入的编号是否存在
        if id not in ids:
            print('您查询的电影不存在')
            return
        else:
            print('编号 名称 评分')
            # 打印该编号的信息
            for i in data[ids.index(id)]:
                print(i, end=' ')
            print()
    else:
        name = input('输入名称')
        # 职工姓名列表
        names = [i[1] for i in data]
        # 判断输入的姓名是否存在
        if name not in names:
            print('您查询的电影不存在')
            return
        else:
            print('编号 名称 评分')
            # 同上
            for i in data[names.index(name)]:
                print(i, end=' ')
            print()

def statistics():
    # 数据准备
    print('评分的平均值：',np.array(data).T[2].astype(float).mean())
    x = list(np.array(data).T[0].astype(int))
    y = list(np.array(data).T[2].astype(float))
    # 使用 Matplotlib 画折线图
    plt.plot(x, y)
    # 设置图表的标题以及x和y轴的说明
    plt.title('评分')
    plt.xlabel('编号ID', fontsize=18)
    plt.ylabel('评分', fontsize=18)
    # 设置刻度标记的文字大小
    plt.tick_params(axis='both', labelsize=16)
    plt.show()
    
if __name__ == '__main__':
    df = pd.read_excel('253.xlsx', index_col=None)
    data = np.array(df).tolist()
    # 读取文件 如果文件不存在 报错跳过 无视
    while True:
        # 根据玩家的输入 选择相应的功能
        choice = int(input('1.添加电影数据\n2.显示电影数据\n3.查询电影数据\n6.统计电影评分数据\n9.退出'))
        if choice == 1:
            add()
        elif choice == 2:
            show()
        elif choice == 3:
            quety()
        elif choice == 6:
            statistics()
        elif choice == 9:
            print('退出程序')
            break

最终效果：

标签：plt,name,Python,查询,评分,print,input,data
来源： https://blog.csdn.net/weixin_54707168/article/details/112797309

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。