python将知识图谱的节点关系（CSV或其他格式）转换成Echarts所需的json格式

python将知识图谱的节点关系（CSV或其他格式）转换成Echarts所需的json格式

前言：

1. 此代码以如下（CSV）格式的数据为例，

故事	时间	地点	人物
xx	2020	安徽合肥	小戈

并处理保存为如

{"data": [{"name": "xx"}, {"name": "2020"}, {"name": "\u5b89\u5fbd\u5408\u80a5"}, {"name": "\u5c0f\u6208"}], "links": [{"source": "2020", "target": "2020", "name": "\u65f6\u95f4"}, {"source": "2020", "target": "\u5b89\u5fbd\u5408\u80a5", "name": "\u5730\u70b9"}, {"source": "2020", "target": "\u5c0f\u6208", "name": "\u4eba\u7269"}]}

的JSON文件（中文默认为unicode编码）

2. Echarts得到数据并引用时，如下直接放入即可

,
data: jsonData.data,
links: jsonData.links

3. 得学会举一反三

import csv
import json

# 开始读取数据
# 开始转换为json格式数据

data_json = "{\"data\": ["
links = ", \"links\": ["
data = []
relationship = ['时间', '地点', '人物']
# 使用test.csv，这只是所有数据的一部分
with open('test.csv', encoding="utf-8") as csvfile:
    csv_reader = csv.reader(csvfile)  # 使用csv.reader读取csvfile中的文件
    data_header = next(csv_reader)  # 读取第一行每一列的标题
    # print(data_header)
    for row in csv_reader:  # 将csv 文件中的数据保存到data_data中
        for i in range(len(data_header)):
            if row[i] not in data and row[i] != '':
                data.append(row[i])
                data_json = data_json + '{\"name\":"' + row[i] + '"},'
        # 至此，我们已经获得了所有节点
        for index, relation in enumerate(relationship):
            links = links + '{\"source\":"' + row[1] + '",\"target\":"' + row[index+1] + '\",\"name\":\"' + relation + '\"},'
        # 至此，我们已经获得了所有关系
    # print(len(data))
    data_json = data_json.rstrip(',')
    data_json += "]"
    # print(data_json)
    # 至此，节点已经转换完成，接下来添加关系
    links = links.rstrip(',')
    links += "]}"
    # print(links)
    data_json += links
    # print(data_json)
    # 至此，我们已经获得了data_json数据
# 开始写入json文件
new_data = json.loads(data_json, strict=False)
# print(new_data)
with open("test_result.json", "w", ) as f:
    json.dump(new_data, f)
# 至此我们已经获得了json数据，嫌格式不好看的话可以去网上自己格式化一下