生成王者荣耀装备信息Markdown和Excel

王者荣耀的装备信息列表网址:https://pvp.qq.com/web201605/item.shtml

生成王者荣耀装备信息Markdown和Excel_第1张图片

经开发者工具检查,装备数据可通过https://pvp.qq.com/web201605/js/item.json接口获取。

获取装备数据

采集并整理数据:

import requests
import pandas as pd

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                  'Chrome/88.0.4324.104 Safari/537.36 '
}
target = 'https://pvp.qq.com/web201605/js/item.json'
item_list = requests.get(target, headers=headers).json()
item_df = pd.DataFrame(item_list)
item_df.sort_values(["item_type", "price", "item_id"], inplace=True)
item_df.fillna("", inplace=True)
item_df.des1 = item_df.des1.str.replace("", "", regex=True)
item_df.des2 = item_df.des2.str.replace("", "", regex=True)
item_df

生成王者荣耀装备信息Markdown和Excel_第2张图片

查看装备图片网址,例如铁剑的网址为:

https://game.gtimg.cn/images/yxzj/img201606/itemimg/1111.jpg

可以发现规律,item_id是图片的名称。

多线程下载图片

import os
from concurrent.futures import ThreadPoolExecutor


def download_img(item_id):
    if os.path.exists(f"imgs/{item_id}.jpg"):
        return
    imgurl = f"http://game.gtimg.cn/images/yxzj/img201606/itemimg/{item_id}.jpg"
    res = requests.get(imgurl)
    with open(f"imgs/{item_id}.jpg", "wb") as f:
        f.write(res.content)


os.makedirs("imgs", exist_ok=True)
with ThreadPoolExecutor(max_workers=8) as executor:
    nums = executor.map(download_img, item_df.item_id)

几秒钟之内已经下载完全部图片:

生成王者荣耀装备信息Markdown和Excel_第3张图片

生成Markdown文档

下面我们生成一下对应的Markdown文档,首先整理一下数据:

item_type_dict = {1: '攻击', 2: '法术', 3: '防御', 4: '移动', 5: '打野', 7: '游走'}
item_ids = item_df.item_id.values
item_df.item_id = item_df.item_id.apply(
    lambda item_id: f"![{item_id}](imgs/{item_id}.jpg)")
item_df.item_type = item_df.item_type.map(item_type_dict)
item_df.columns = ["图片", "装备名称", "类型", "售价", "总价", "基础描述", "扩展描述"]
item_df

生成王者荣耀装备信息Markdown和Excel_第4张图片

按装备类型分组,生成Markdown文档:

with open("王者装备说明.md", "w") as f:
    for item_type, item_split in item_df.groupby("类型", sort=False):
        f.write(f"# {item_type}\n")
        item_split.drop(columns="类型", inplace=True)
        f.write(item_split.to_markdown(index=False))
        f.write("\n\n")

生成的Markdown如下:

生成王者荣耀装备信息Markdown和Excel_第5张图片

不过Markdown的表格无法任意调整,图片需要点击后才会放大,下面我们考虑生成Excel表格:

生成Excel

首先整理数据:

item_df.图片 = ""
item_df.基础描述 = item_df.基础描述.str.replace("
"
, "\n") item_df.扩展描述 = item_df.扩展描述.str.replace("
"
, "\n") item_df

生成王者荣耀装备信息Markdown和Excel_第6张图片

然后写出来Excel表:

from openpyxl.drawing.image import Image
from openpyxl.styles import Alignment

with pd.ExcelWriter("王者装备说明.xlsx", engine='openpyxl') as writer:
    item_df.to_excel(writer, sheet_name='装备说明', index=False)
    worksheet = writer.sheets['装备说明']
    worksheet.column_dimensions["A"].width = 11
    for item_id, (cell,) in zip(item_ids, worksheet.iter_rows(2, None, 1, 1)):
        worksheet.row_dimensions[cell.row].height = 67
        worksheet.add_image(Image(f"imgs/{item_id}.jpg"), f'A{cell.row}')
    worksheet.column_dimensions["F"].width = 15
    worksheet.column_dimensions["G"].width = 35
    writer.save()

生成结果如下:

生成王者荣耀装备信息Markdown和Excel_第7张图片

经过一番人工调整后:

生成王者荣耀装备信息Markdown和Excel_第8张图片

大家还可以根据自己想要的效果生成HTML和Word等等。

你可能感兴趣的:(数据处理,python)