用python绘制关系矩阵

Question

我有一个使用 wxpython 的 python 界面，它允许用户填写一个矩阵（0/1），然后为他们绘制图表。该程序创建一个 numpy 矩阵，然后根据该矩阵创建一个 networkx 图，然后使用 matplotlib.pylab 显示该图。

numpy 是必须的，因为该程序还可以执行其他操作，例如获取传递、自反和对称闭包...至于 networkx，如果您推荐其他更好的用于绘制矩阵的东西，我可以使用其他东西，至于 matplotlib，我讨厌如果您知道显示图表的任何其他方式，请提出建议。

matplotlib 是我的问题的根源，当用户单击图形按钮时，我的程序读取矩阵，制作图形，然后 matplotlib 将其显示在新窗口中（默认情况下）。现在，如果用户在没有先关闭 matplotlib 窗口的情况下返回到原始窗口并绘制不同的矩阵，程序就会崩溃。

在我看来，绘制关系“箭头”的方式也没有吸引力。

我需要一种更好的方法来绘制矩阵，或者至少作为强制关闭 myplotlib 窗口的一种方法，我尝试了 plt.close() 但这不起作用，窗口将保持打开状态，并且两个窗口都会说（不是响应），我必须结束进程。

这是有问题的代码的一部分：

import numpy as np
import networkx as nx
import matplotlib.pylab as plt

... ... ...

def graph(values)
    plt.close()      #with or without this it does not work
    matrix = np.matrix(values)
    graph = nx.DiGraph(matrix)
    nx.draw(graph)
    plt.show()
    return

Answer 1

在我看来，您的主要抱怨是 wx 处理 matplotlib 窗口的方式。可以将 Matplotlib 图形嵌入到您的 wx 窗口中。这是一个例子：

http://wiki.scipy.org/Matplotlib_figure_in_a_wx_panel （已更新）

它变得有点复杂。基本上，您应该复制代码并替换“DemoPlotPanel.draw()”方法。您需要修改代码以指定要绘制的轴。它隐藏在此处的networkx文档中：

http://networkx.lanl.gov/reference/drawing.html

Answer 2

我只是使用 Networkx 文档的一个示例：

try:  
    import matplotlib.pyplot as plt  
except:  
    raise  

import networkx as nx  

G=nx.star_graph(20)  
pos=nx.spring_layout(G)  
nx.draw(G,pos)  
plt.show() # display

所以你一开始的 plt.close() 语句没有意义，你应该删除它。你还应该计算节点的坐标，句子 pos=nx.spring_layout( G) 这样做。您调用特定的布局算法并提供图 G，您将获得一个字典作为回报，其中包含每个节点的 x 和 y 坐标。

仔细查看以下示例：http://networkx.lanl.gov/gallery.html