您正确尝试使用nx.draw_networkx_edge_labels
。但它使用labels
as edge_labels
并且您没有在任何地方指定它。你应该创建这个字典:
labels = {e: G.edges[e]['score'] for e in G.edges}
并取消注释nx.draw_networkx_edge_labels
功能:
import networkx as nx
import pandas as pd
import matplotlib.pyplot as plt
feature_1 = ['Boston', 'Boston', 'Chicago', 'ATX', 'NYC']
feature_2 = ['LA', 'SFO', 'LA', 'ATX', 'NJ']
score = ['1.00', '0.83', '0.34', '0.98', '0.89']
df = pd.DataFrame({'f1': feature_1, 'f2': feature_2, 'score': score})
print(df)
G = nx.from_pandas_edgelist(df=df, source='f1', target='f2', edge_attr='score')
pos = nx.spring_layout(G, k=10) # For better example looking
nx.draw(G, pos, with_labels=True)
labels = {e: G.edges[e]['score'] for e in G.edges}
nx.draw_networkx_edge_labels(G, pos, edge_labels=labels)
plt.show()
所以结果将如下所示:
附:您的源/目标也不正确nx.from_pandas_edgelist
。你应该有:
source='f1', target='f2'
代替:
source='feature_1', target='feature_2'