python——地震数据统计A

地震数据统计A

类型：文件

描述

附件文件中包含若干条近年的地震相关信息，请实现以下功能完成数据统计，输入输出参考示例：‪‬‪‬‪‬‪‬‪‬‮‬‪‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‪‬

示例 1

输入：

震源深度 
3  

输出：

从低到高前3名:
吉林延边州珲春市(疑爆):0千米
江苏连云港市灌南县(疑爆):0千米
辽宁抚顺市新抚区（矿震）:0千米

从高到低前3名:
斐济群岛地区:0千米
斐济群岛地区:620千米
斐济群岛地区:610千米

示例 2

输入：

震级
3

输出：

斐济群岛地区:8.1级
秘鲁北部:7.8级
斐济群岛地区:7.8级

示例 3

输入：

印度洋

输出：

东南印度洋海岭:5.7
东南印度洋:6.2
中印度洋海岭:5.9
西南印度洋海岭:5.3
西南印度洋海岭:5.9‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬‬

参考代码

def type_judge(input_str):
    """接收一个字符串为参数，根据参数调用不同的函数进行运算。
    """
    if input_str == '震源深度':
        depth_of_focus(data)
    elif input_str == '震级':
        magnitude(data)
    else:
        others(data)


def read_file(filename):
    """读文件，返回二维列表"""
    with open(filename, 'r', encoding='UTF-8') as f:
        data_lst = [i.strip().split(',') for i in f]
    return data_lst[1:]


def depth_of_focus(data_lst):
    n = int(input())
    data_sort_asc = sorted(data_lst, key=lambda x: eval(x[-2]))
    print('从低到高前{}名:'.format(n))
    for i in data_sort_asc[:n]:
        print('{}:{}千米'.format(i[-1], i[-2]))
    print()

    print('从高到低前{}名:'.format(n))
    data_sort_desc = sorted(data_lst, key=lambda x: eval(x[-2]),reverse=True)
    for i in data_sort_desc[:n]:
        print('{}:{}千米'.format(i[-1], i[-2]))


def magnitude(data_lst):
    n = int(input())
    data_sort = sorted(data_lst, key=lambda x: eval(x[1]), reverse=True)
    for i in data_sort[:n]:
        print('{}:{}级'.format(i[-1], i[1]))


def others(data_lst):
    """不是上述情况，在地名中查找输入的字符串，如果存在，则输出该地区数据"""
    flag = 0  # 如果不存在，则输出无数据
    for line in data_lst:
        if question in line[-1]:
            print('{}:{}'.format(line[-1], line[1]))
            flag = 1
    if flag == 0:
        print('无数据')


if __name__ == '__main__':
    file = 'quake.csv'
    data = read_file(file)
    question = input()
    type_judge(question)

本人代码：

with open('quake.csv', 'r', encoding='utf-8') as f:
    data = [one.strip().split(',') for one in f.readlines()[1:]]

n = input()
if n == '震源深度':
    num = int(input())
    data.sort(key=lambda x: eval(x[-2]), reverse=False)
    print('从低到高前{}名:'.format(num))
    for i in range(num):
        print('{}:{}千米'.format(data[i][-1], data[i][-2]))
    data.sort(key=lambda x: eval(x[-2]), reverse=True)
    print()
    print('从高到低前{}名:'.format(num))
    for i in range(num):
        print('{}:{}千米'.format(data[i][-1], data[i][-2]))
elif n == '震级':
    num = int(input())
    data.sort(key=lambda x: x[1], reverse=True)
    for i in range(num):
        print('{}:{}级'.format(data[i][-1], data[i][1]))
else:
    flag = 0
    for one in data:
        if n in one[-1]:
            print('{}:{}'.format(one[-1], one[1]))
            flag = 1
    if flag == 0:
        print('无数据')