2017-06-13 41 views
0

nxpd的文檔非常清楚如何打印ipython內部的圖形,但是如何將圖形保存爲.png或.pdf?下面的方法讀入包含狀態更改概率的數據幀(mc)和保存繪製圖的輸出路徑。我不清楚如何將nxpd繪製傳遞給pyplot或使用內部保存方法。保存nxpd生成的圖形

NXPD文檔是有限的:https://github.com/chebee7i/nxpd

def DrawMarkov(mc, outputPath): 
    import matplotlib.pyplot as plt 
    from pandas import DataFrame 
    import networkx as nx 
    from nxpd import draw 
    ## Draw as an internal to ipython 
    # draw(G, show='ipynb') 
    ## OR 
    # from nxpd import nxpdParams 
    # nxpdParams['show'] = 'ipynb' 

    def Add_Nodes(G, name): 
     for node in range(0,len(mc)): 
    #   print(node); print(type(node)); print(color_map[node]) 
      G.add_node(node, style='filled', fillcolor=color_map[node],) 

    def Add_Edges(G,mc): 
     for start in range(0,len(mc[0])): 
      # print(start,'\n') 
      for finish in range(0,len(mc)): 
       # print(mc[start][finish]) 
       if mc[start][finish]!=0: 
        G.add_edge(start, finish, weight=mc[start][finish], label=str(mc[start][finish])) 

    color_map = {0: 'lightblue', 
       1:'palegreen', 
       2:'lightcoral'} 

    ## Plot DiGraph 
    G = nx.DiGraph() 
    Add_Nodes(G,mc) 
    Add_Edges(G,mc) 
    mapping={0:'State1', 1:'State2', 2:'State3'} 
    G = nx.relabel_nodes(G,mapping) 

    ## Output 
    draw(G, layout='dot') # , show='ipynb' # unclear how to pass graph to plt 
    # plt.savefig(os.path.join(outputPath, 'Graph.png')) 
+0

會發生什麼?假設它就像networkx一樣,應該這樣做。 – Joel

+0

如果未使用show ='ipynb'參數,則draw()會在新的預覽窗口中生成.png(我在Mac上)。 plt.savefig('graph.png')產生:「AttributeError:'str'對象沒有屬性'savefig'」。 – MyopicVisage

+0

我將如何壓制.png文件的打開並強制它保存? – MyopicVisage

回答

-1

您可以指定在繪製函數保存方式。有關詳細信息,請參閱實施https://github.com/chebee7i/nxpd/blob/master/nxpd/nx_pydot.py

用法:

draw(G, format ='pdf') 

支持的格式包括:如果你只是做`plt.savefig( 'Graph.png')`

'canon', 'cmap', 'cmapx', 'cmapx_np', 'dia', 'dot', 
'fig', 'gd', 'gd2', 'gif', 'hpgl', 'imap', 'imap_np', 
'ismap', 'jpe', 'jpeg', 'jpg', 'mif', 'mp', 'pcl', 'pdf', 
'pic', 'plain', 'plain-ext', 'png', 'ps', 'ps2', 'svg', 
'svgz', 'vml', 'vmlz', 'vrml', 'vtx', 'wbmp', 'xdot', 'xlib' 
+0

儘管此鏈接可能會回答問題,但最好在此處包含答案的基本部分,並提供供參考的鏈接。如果鏈接頁面更改,則僅鏈接答案可能會失效。 - [來自評論](/ review/low-quality-posts/17465409) –

+0

嗨@MadPhysicist,我在我的答案中包含了使用示例。感謝whrrgarbl的編輯,使之清楚。 –