python - 使用 Python 的 `daft` 库在图形图中添加自边缘

Question

我想在下图中添加一条将 beta 链接到 beta 的边：

使用DAFT生成以下代码：

from matplotlib import rc
rc("font", family="serif", size=12)
rc("text", usetex=True)
import daft

pgm = daft.PGM([2.3, 3.05], origin=[0.3, 0.3], observed_style="inner")

# Hierarchical parameters.
pgm.add_node(daft.Node("beta", r"$\beta$", 1.5, 2))

# Latent variable.
pgm.add_node(daft.Node("w", r"$w_n$", 1, 1))

# Data.
pgm.add_node(daft.Node("x", r"$x_n$", 2, 1, observed=True))

# Add in the edges.
pgm.add_edge("beta", "beta")  # Attempting to create a self-edge, but no effect!
pgm.add_edge("w", "x")
pgm.add_edge("w", "w")
pgm.add_edge("w", "beta")
pgm.add_edge("beta", "x")

# Render and save.
pgm.render()
pgm.figure.savefig("nogray.pdf")

但为什么它不起作用？尤其是这条线pgm.add_edge("beta", "beta")。我欢迎 Daft 以外的其他建议，只要它在 Python 下。

Answer 1

如果您从第 301 行（类）查看源代码，Edge您将看到所有线都是基于坐标的直线给出，([x, x + dx], [y, y + dy])如下面的无向边代码所示：

x, y, dx, dy = self._get_coords(ctx)

# Plot the line.
line = ax.plot([x, x + dx], [y, y + dy], **p)
return line 

因此，似乎没有一种定义自边缘的方法（因为这样的边缘需要弯曲才能弯曲回同一节点）。

至于您可能想要查看的替代库networkx，以下文档显示了 with 的使用self loops。DAFT或者，您可以在Github上提出问题。