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

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python修改代码实现添加，显示，查询和统计电影评分数据，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2560字，纯文字阅读大概需要4分钟。

内容图文

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

通过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

最终效果：

内容总结

以上是互联网集市为您收集整理的Python修改代码实现添加，显示，查询和统计电影评分数据全部内容，希望文章能够帮你解决Python修改代码实现添加，显示，查询和统计电影评分数据所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端