self-constructing_graph/agent/testnx.py

import networkx as nx
import pandas as pd
import json
entity_data = pd.DataFrame({"id":[1,1,2,3],"name":['a','aa','b','c'], "type":['x','x','y','z']})
entity_data.set_index("id", inplace=True)
#创建测试的节点数据
jsondata = '''
[
    [
        1291243,
        {
            "name": "好急的胃肠炎"
        }
    ]
]'''
print(json.loads(jsondata))
relation_data = pd.DataFrame({"src":[1,2], "dest":[2,3], "type":['normal','normal'], "weight":[1,1]})

results = entity_data[entity_data.index==5]
print(results.empty)
results = results.iloc[0]
print(results['name'])
results['name'] ='A'
print(results['name'])
results = entity_data[entity_data.index==1]
for index,data in results.iterrows():
    print(index,data['name'],data['type'])
print("----")
entity_data.loc[1, 'type'] = 'unknown'
entity_data.loc[(entity_data['name']=='b') & (entity_data.index == 2), 'type'] = 'unknown other'
for index,data in entity_data.iterrows():
    print(index,data['name'],data['type'])
print("----")
print(results)
# G = nx.from_pandas_edgelist(relation_data, "src", "dest", edge_attr=True, create_using=nx.DiGraph())
# nx.set_node_attributes(G, entity_data.set_index("id").to_dict("index"))

# print(G.nodes(data=True))
# print(G.edges(data=True))
# print(list(G.neighbors(1)))

# print(G.edges(2.0))