送书 | 用 Neo4j 理一理王者荣耀英雄之间千丝万缕的关系,挺好玩!-技术圈

本次送书之前先给大家介绍我的好友周萝卜，他是一个非常有趣的人，经常使用python做一些骚操作！今天这篇文章就是他给大家带来的分享。下面是他的公众号，大家感兴趣的可以关注一下他哦！

正文开启

当我们进入王者荣耀游戏客户端，是可以在英雄关系页面看到不同英雄之间的关联信息的，比如花木兰和兰陵王之间的宿命，安其拉和亚瑟的单恋以及露娜与铠那命运弄人的兄妹之情等等

要厘清这些英雄之间的关系，还是比较困难的，尤其是还有很多中心英雄，与其有关系的英雄多大数十个。今天我们就通过 neo4j 数据库来处理这些错综复杂的关系，看看一直伴随我们每个王者峡谷的英雄都有着怎样的“朋友圈”呢

数据获取

找了很久，还是在网上找到了一个比较全面的英雄关系数据

https://www.haosix.com/gonglue/411

网页很简单，直接上 requests 爬取即可

import requests
from bs4 import BeautifulSoup
import pandas as pd
import os
import csv


res = requests.get("https://www.haosix.com/gonglue/411")

然后我们通过 beautifulsoup 来解析网页

soup = BeautifulSoup(res.text)
div = soup.find('div', class_="bbcode-body bbcode-body-v2 markdown-body")
p_list = div.find_all("p")[2:-1]

现在列表 p_list 当中就是我们需要的数据了

接下来我们编写一个处理数据的函数，把网页上的数据转换成我们需要的形式

def deal_str(data):
    rel_dict = {}
    name = data.split("：")
    name1 = name[0]
    name_rel = name[1].split("、")
    if r"(" in name_rel[0]:
        for i in name_rel:
            tmp = i.split("(")
            rel_dict[tmp[0]] = tmp[1].split(")")[0]
        return [name1, rel_dict]
    return [name1, "无"]

我们来看下函数效果

deal_str('裴擒虎：杨玉环(女神与同伴)、公孙离(收留之人与同伴)、弈星(互补的同伴)、明世隐(下属与首领)、苏烈(下属与曾经上司)')

得到了英雄名称以及和他有关系的英雄关系信息

再下来就是保存到 csv 文件

def save_data_name(data_name, list_info):
    if not os.path.exists(data_name + r'.csv'):
        # 表头
        name = ["name","id"]
        # 建立DataFrame对象
        file_test = pd.DataFrame(columns=name, data=list_info)
        # 数据写入
        file_test.to_csv(data_name + r'.csv', encoding='utf-8', index=False)
    else:
        with open(data_name + r'.csv', 'a+', newline='', encoding='utf-8') as file_test:
            # 追加到文件后面
            writer = csv.writer(file_test)
            # 写入文件
            writer.writerows(list_info)

一切准备工作完毕之后，就可以整合代码了

content = {}
index = 0
for p in p_list:
    content = deal_str(p.text)
    save_data_name("name", [[content[0], index]])
    if type(content[1]) == dict:
        for k,v in content[1].items():
            print([content[0], k, v])
            save_data_relation("relation", [[content[0], k, v]])
    index += 1

处理过程如下

这要我们就得到了两个 csv 文件，一个保存了英雄名称，另一个保存了英雄关系信息

Neo4j 使用简介

Neo4j 是目前最为流行的图数据库，用于存储丰富的关系数据。图是由顶点（Vertex），边（Edge）和属性（Property）组成的，顶点和边都可以设置属性，顶点也称作节点，边也称作关系，每个节点和关系都可以由一个或多个属性

对于 Neo4j 的安装就不再赘述了，想我这里就是在自己本地的 Windows 上安装了一个，还是比较简单的

安装完成之后，我们一般可以在本地开发 Neo4j 控制浏览器

http://localhost:7474/browser/

然后我们把前面生成的两个文件放到 Neo4j 安装目录的 import 文件夹下（安全考虑，Neo4j 默认只能从该目录下导入文件）

接下来我们在 Neo4j 命令行中编写 Cypher 命令，先导入 csv 文件
英雄名称文件：

LOAD CSV WITH HEADERS FROM "file:///name.csv" AS line MERGE (p:person{name:line.name,id:line.id})

英雄关系文件：

LOAD CSV WITH HEADERS FROM "file:///relation.csv" AS relations MATCH (entity1:person{name:relations.name1}), (entity2: person{name:relations.name2}) CREATE (entity1)-[:rel{relation:relations.relation}]->(entity2)