构建金融知识图谱
from sklearn import metrics
import tushare as ts
from sklearn.cluster import KMeans
import numpy
import collections
import pandas
from sklearn import metrics
import pandas as pd
import time
from multiprocessing import Pool
from py2neo import Graph,Node,Relationship
##连接neo4j数据库,输入地址、用户名、密码
graph = Graph('http://localhost:7474',username='neo4j',password='08300734')
#获取连接备用
cons = ts.get_apis()
pro = ts.pro_api()
pp=pro.daily_basic(ts_code='', trade_date='20181105')
code=list(pp['ts_code'])
ccpt = pro.concept()
df = pro.concept_detail(id='TS2', fields='ts_code,name')
data = pro.query('stock_basic', exchange='', list_status='L', fields='ts_code,symbol,name,area,industry,list_date')
data1=data[['ts_code','name']]
test_node_2 = Node(label='ru_yi_zhuan',name='皇帝')
data2=data1.values
for m in ccpt[['code','name']].values[:100]:
fd1=m[0]
pd1=m[1]
dfd = pro.concept_detail(id=fd1, fields='ts_code,name')
dfd1=dfd.values
ctt=Node(label='concept',name=pd1)
graph.create(ctt)
for tt in dfd1:
fdd=tt[1]
sk = Node(label='stock',name=fdd)
node_munv_node = Relationship(sk,'belong_to_concept',ctt)
graph.create(sk)
graph.create(node_munv_node)
print(fdd,pd1)
建立股票node 和概念node ,并且指定关系
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY