python - 无法使用来自 networkx 的 from_pandas_dataframe 创建有向图

Question

我正在学习 networkx 库并使用 twitter 转发有向图数据。我首先将数据集读入 pandas df(列为“from”、“to”、“weight”)，并想使用以下代码将前 300 行(转发)放入图表中:

tw_small = nx.from_pandas_dataframe(edges_df[:300],source='from',
                                   target='to',edge_attr=True)

我认为它正确地创建了一个图，但是当我运行 tw_small.is_directed() 时，它显示 False(无向图)，我使用 绘制了一个图>nx.draw() 但它也不显示方向。

有人能帮我找到制作有向图的正确方法吗？

谢谢。

Answer 1

添加可选的关键字参数 create_using=nx.DiGraph(),

tw_small = nx.from_pandas_dataframe(edges_df[:300],source='from',
                                   target='to',edge_attr=True,
                                   create_using=nx.DiGraph())