文件操作之extract_weapon

sanguoyanyi.txt文档的准备，详见文件操作之extract_name

下面是weapon.txt的内容：

练习：提取相关武器，并统计出现的次数，然后排序

import re


def find_item2(weapon):
    with open('sanguoyanyi.txt', encoding='utf-8') as f1:
        data = f1.read().replace('
', '')
        weapon_num = len(re.findall(weapon, data))
        print("33[31:1m%s33[0m出现了%s 次" % (weapon, weapon_num))
    return weapon_num


weapon_dict = {}
# with语句叫上下文管理器，会自动关闭被调用打开的文件
with open('weapon.txt', encoding='utf-8') as f:
    i = 1
    for line in f:
        if i % 2 == 1:
            weapon_name = line.strip()
            weapon_dict[weapon_name] = find_item2(weapon_name)
        i += 1
weapon_sorted1 = sorted(weapon_dict.items(), key=lambda item: item[1], reverse=True)
print('按次数降序排列：', weapon_sorted1)
weapon_sorted2 = sorted(weapon_dict.items())
print('按武器排列：', weapon_sorted2)