6.1 葡萄酒评论分析报告

任务描述

本关任务：编写程序，多维度分析葡萄酒数据。

第1关 葡萄酒评论分析报告——国家列表和平均分

代码如下：

import pandas as pd

# 定义符号常量，用于索引，使之具有清晰的语义
COUNTRY = 1
POINTS = 3


def csv_to_ls(file):
    """接收文件名为参数，用pandas读取文件中的数据，数据部分转为二维列表类型，返回二维列表。"""
    wine_list = pd.read_csv(file).values.tolist()
    return wine_list


def country_ls(wine_list):
    """接收列表格式的葡萄酒数据为参数，略过标题行，返回不重复的国家名列表，按字母表升序排序，
    若国家名数据缺失，略过该条数据，返回值中不包含空字符串元素。
    @参数 wine_list：葡萄酒数据，列表类型
    """
    # 此处补充你的代码
    country_list = []
    for x in wine_list:
        if x[COUNTRY] not in country_list:
            country_list.append(x[COUNTRY])
    country_list.sort()
    return country_list
 

def avg_point(wine_list, country):
    """接收列表格式的葡萄酒数据和国家名列表为参数，计算每个国家的葡萄酒的平均得分，
    返回值为国家名和得分的列表。
    @参数 wine_list：葡萄酒数据，列表类型
    @参数 country：国家名，列表类型
    """
    # 此处补充你的代码
    score=[]
    for i in range(len(country)):
        n=0
        count=0
        for j in wine_list:
            if country[i]==j[1]:
                n+=float(j[-3])
                count+=1
        score.append([country[i],round(n/count,2)])
    ls=sorted(score,key=lambda x:x[0], reverse=False)
    return ls

def judge(txt):
    """接收一个字符串为参数，根据参数值调用不同函数完成任务"""
    filename = 'data/winemag-data.csv'
    wine = csv_to_ls(filename)
    country = country_ls(wine)
    if txt == '国家名列表':
        print(country)
    elif txt == '平均分':
        print(avg_point(wine, country))  # 每个国家的葡萄酒的平均得分
    else:
        print('输入错误')


if __name__ == '__main__':
    text = input()
    judge(text)

第2关 葡萄酒评论分析报告——平均分排序和评分最高

代码如下：

import pandas as pd
import math

# 定义符号常量，用于索引，使之具有清晰的语义
NUMBER = 0
COUNTRY = 1
DESCRIPTION = 2
POINTS = 3
PRICE = 4


def csv_to_ls(file):
    """接收文件名为参数，用pandas读取文件中的数据，数据部分转为二维列表类型，返回二维列表。"""
    wine_list = pd.read_csv(file).values.tolist()
    return wine_list


def country_ls(wine_list):
    """接收列表格式的葡萄酒数据为参数，略过标题行，返回不重复的国家名列表，按字母表升序排序，
    若国家名数据缺失，略过该条数据，返回值中不包含空字符串元素。
    @参数 wine_list：葡萄酒数据，列表类型
    """
    country_list = []
    for x in wine_list:
        if x[COUNTRY] not in country_list:
            country_list.append(x[COUNTRY])
    country_list.sort()
    # print(country_list)
    return country_list


def avg_point_sort(wine_list, country):
    """接收列表格式的葡萄酒数据和国家名列表为参数，计算每个国家的葡萄酒的平均得分，
    返回值为国家名和得分的列表，按评分由高到低降序排列。
    @参数 wine_list：葡萄酒数据，列表类型
    @参数 country：国家名，列表类型
    """
    # 此处补充你的代码
    score=[]
    for i in range(len(country)):
        n=0
        count=0
        for j in wine_list:
            if country[i]==j[1]:
                n+=float(j[-3])
                count+=1
        score.append([country[i],round(n/count,2)])
    ls=sorted(score,key=lambda x:x[1], reverse=True)
    return ls


def top_10_point(wine_list):
    """接收列表格式的葡萄酒数据参数，返回评分最高的十款葡萄酒的编号、出产国、评分和价格，按评
    分降序输出。
    需要注意的是评分可能有缺失值，此时该数据为nan
    if math.isnan(x) == False可用于判定x的值是不是nan
    nan的数据类型是float,不可以直接用字符串判定方法。
    @参数 wine_list：葡萄酒数据，列表类型
    """
    # 此处补充你的代码
    list1=[]
    for i in wine_list:
        if math.isnan(float(i[-3]))==False:
            list1.append(i)
    list2=sorted(list1,key=lambda x:x[-3],reverse=True)
    list3=[]
    count=1
    for k in list2:
        if count<11:
            list3.append([k[-6],k[-5],k[-3],k[-2]])
            count+=1
    return list3


def judge(txt):
    """接收一个字符串为参数，根据参数值调用不同函数完成任务"""
    filename = 'data/winemag-data.csv'
    wine = csv_to_ls(filename)
    country = country_ls(wine)
    if txt == '平均分排序':
        print(avg_point_sort(wine, country))  # 每个国家的葡萄酒的平均得分降序输出
    elif txt == '评分最高':
        print(top_10_point(wine))  # 评分最高的十款葡萄酒的编号、出产国、评分和价格，按评分降序输出
    else:
        print('输入错误')


if __name__ == '__main__':
    text = input()
    judge(text)

第3关 葡萄酒评论分析报告——价格最高和葡萄酒评分

代码如下：

import pandas as pd
import math

# 定义符号常量，用于索引，使之具有清晰的语义
NUMBER = 0
COUNTRY = 1
POINTS = 3
PRICE = 4


def csv_to_ls(file):
    """接收文件名为参数，用pandas读取文件中的数据，数据部分转为二维列表类型，返回二维列表。"""
    wine_list = pd.read_csv(file).values.tolist()
    return wine_list


def top_20_price(wine_list):
    """接收列表格式的葡萄酒数据参数，返回价格最高的二十款葡萄酒的编号、出产国、评分和价格，按价
    格降序输出。
    @参数 wine_list：葡萄酒数据，列表类型
    需要注意的是价格可能有缺失值，此时该数据为nan
    if math.isnan(x) == False可用于判定x的值是不是nan
    nan的数据类型是float,不可以直接用字符串判定方法。
    """
    # 此处补充你的代码
    list1=[]
    for i in wine_list:
        if math.isnan(float(i[-2]))==False:
            list1.append(i)
    list2=sorted(list1,key=lambda x:x[-2],reverse=True)
    list3=[]
    count=1
    for k in list2:
        if count<21:
            list3.append([k[-6],k[-5],k[-3],k[-2]])
            count+=1
    return list3

def amount_of_point(wine_list):
    """接收列表格式的葡萄酒数据参数，返回每个评分的葡萄酒数量，忽略没有评分的数据
    例如[...[84, 645], [85, 959],...]表示得分为84的葡萄酒645种，得分85的葡萄酒有959种。
    @参数 wine_list：葡萄酒数据，列表类型
    """
    # 此处补充你的代码
    score_list = []
    list1 = []
    count = 0
    for x in wine_list:
        if x[-3] not in score_list:
            score_list.append(x[-3])
    score_list.sort()
    for i in score_list:
        count = 0
        for j in wine_list:
            if i == int(j[-3]):
                count += 1
        list1.append([i, count])
    return list1


def most_of_point(amount_of_points):
    """接收每个评分的葡萄酒数量的列表为参数，返回获得该分数数量最多的评分和数量的列表。
    @参数 amount_of_points：每个评分的葡萄酒数量，列表类型
    """
    # 此处补充你的代码
    list1 = sorted(amount_of_points, key=lambda x: x[1], reverse=True)
    return list1[0]


def avg_price_of_most_point(wine_list, most_of_points):
    """接收列表格式的葡萄酒数据和获得最多的评分及数量的列表为参数
    忽略缺失价格的数据，返回这个分数的葡萄酒的平均价格，保留2位小数。
    @参数 wine_list：葡萄酒数据，列表类型
    @参数 most_of_points：获得最多的评分及数量，列表类型
    """
    # 此处补充你的代码
    sum = 0
    m = 0
    for i in wine_list:
        if most_of_points[0] == int(i[-3]):
            if math.isnan(float(i[-2])) == False:
                sum += float(i[-2])
                m += 1
    return round(sum / m, 2)


def judge(txt):
    """接收一个字符串为参数，根据参数值调用不同函数完成任务"""
    filename = 'data/winemag-data.csv'
    wine = csv_to_ls(filename)
    if txt == '价格最高':
        print(top_20_price(wine))  # 价格最高的二十款葡萄酒的编号、出产国、评分和价格，按价格降序输出
    elif txt == '葡萄酒评分':
        amount_point = amount_of_point(wine)
        most_point = most_of_point(amount_point)
        print(amount_point)  # 各个评分的葡萄酒数量
        print(most_point)  # 拥有葡萄酒数量最多的评分和数量
        print(avg_price_of_most_point(wine, most_point))  # 拥有葡萄酒数量最多的评分的葡萄酒的平均价格
    else:
        print('输入错误')


if __name__ == '__main__':
    text = input()
    judge(text)

尝试静下心来做自己该做的的事情，不要让焦虑和烦恼毁掉你本就不多的热情与定力。