任务描述

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

第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)

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