python对GDELT生成简单知识图谱概念图

BruceNeeal

已于 2023-10-18 23:28:27 修改

阅读量389

点赞数 1

分类专栏： Python Data Mining 文章标签： python 知识图谱开发语言

于 2023-10-12 21:55:55 首次发布

本文链接：https://blog.csdn.net/qq_38195687/article/details/133799700

版权

用python对GDELT events数据中的actor1Name、actor2Name、eventCode字段生成简单的知识图谱图概念图并可视化。

import numpy as np
import pandas as pd
import networkx as nx #图网络包
import matplotlib.pyplot as plt #绘图的包
from mpl_toolkits.mplot3d import Axes3D #3D绘图的包

#代码运行前需自行前往GDELT官网下载数据
data = pd.read_csv(r'E:/WorkSpace/data/GDELT2.0/20231007081500.export.csv', 
	delimiter='\t', encoding='utf-8', low_memory=False, 
	header=None, dtype=str) #保留表头, 读取类型为str
source = []
target = []
relation = []
for line in data.values:
    actor1_name = line[6]
    event_code = line[26]
    actor2_name = line[16]
    if(pd.isna(actor1_name) | pd.isna(event_code) | pd.isna(actor2_name)):
        continue #过滤掉有nan值的数据
    source.append(actor1_name)
    target.append(actor2_name)
    relation.append(event_code)

kg_df = pd.DataFrame({
   'source':source, 'target':target, 'relation':relation