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数据,嫌格式不好看的话可以去网上自己格式化一下

上一篇:python 读取csv文件 和 UnicodeDecodeError: 'utf-8' codec can't decode byte 0xa8 in position


下一篇:vue3 二次封装element-plus 表格组件