1. 程式人生 > >用Python的networkx繪製精美網路圖

用Python的networkx繪製精美網路圖

最近因為數學建模3天速成Python,然後做了一道網路的題,要畫網路圖。在網上找了一些,發現都是一些很基礎的醜陋紅點圖,並且關於網路的一些演算法也沒有講,於是自己進http://networkx.github.io/學習了一下。以下僅博主自己的總結,勿認真,有錯誤盡情指出,大家一起交流。
需要用到的module    malplotlib.pyplot 和networkx 

正文:

一、malplotlib和networkx的安裝(作者使用的是python2.7 pycharm)

在Python的資料夾目錄下Scripts目錄中,如果有pip.exe 檔案,那麼可以用cmd 進入這個目錄,然後輸入  `pip insall --pre matplotlib`直接下載。如果有easy_install 也可以輸入`easy_install。如果都不行就去官網  https://pypi.python.org/pypi/matplotlib/1.5.3`找對應版本下載。
至於networkx,pycharm的porject interpreter裡新增就好。

二、建立圖

networkx有四種圖 Graph 、DiGraph、MultiGraph、MultiDiGraph,分別為無多重邊無向圖、無多重邊有向圖、有多重邊無向圖、有多重邊有向圖。

   import network as nx 
   G = nx.Graph()#建立空的網路圖
   G = nx.DiGraph()
   G = nx.MultiGraph()
   G = nx.MultiDiGraph()

然後是加點和邊了,有多種方法

G.add_node('a')#新增點a
G.add_node(1,1)#用座標來新增點
G.add_edge('x','y'
)#新增邊,起點為x,終點為y G.add_weight_edges_from([('x','y',1.0)])#第三個輸入量為權值 #也可以 list = [[('a','b',5.0),('b','c',3.0),('a','c',1.0)] G.add_weight_edges_from([(list)])

然後是圖形的顯示

#需要匯入matplotlib
import matplotlib.pyplot as plt


nx.draw(G)
plt.show()

為了讓圖形更精美我們詳解nx.draw()

nx.draw(G,pos = nx.random_layout(G),node_color = 'b'
,edge_color = 'r',with_labels = True,font_size =18,node_size =20)

pos 指的是佈局 主要有spring_layout , random_layout,circle_layout,shell_layout。node_color指節點顏色,有rbykw ,同理edge_color.
with_labels指節點是否顯示名字,size表示大小,font_color表示字的顏色。
看到這裡,各位應該就能畫出大量網站上的基本networkx簡單教程了,大概是這個樣子
圖1

三、繪製精美的圖

如果你想要的圖是這樣的
這裡寫圖片描述
或是這樣的
這裡寫圖片描述
還是這樣的
這裡寫圖片描述
這裡寫圖片描述
可以繼續看下去
首先要掌握兩個方法

def draw_networkx_edges(G, pos,
                        edgelist=None,
                        width=1.0,
                        edge_color='k',
                        style='solid',
                        alpha=1.0,
                        edge_cmap=None,
                        edge_vmin=None,
                        edge_vmax=None,
                        ax=None,
                        arrows=True,
                        label=None,
                        **kwds):
G:圖表
   一個networkx圖
pos:dictionary
   將節點作為鍵和位置作為值的字典。
   位置應該是長度為2的序列。

edgelist:邊緣元組的集合
   只繪製指定的邊(預設= G.edges())

width:float或float陣列
   邊線寬度(預設值= 1.0)

edge_color:顏色字串或浮點陣列
   邊緣顏色。可以是單顏色格式字串(default ='r'),
   或者具有與edgelist相同長度的顏色序列。
   如果指定了數值,它們將被對映到
   顏色使用edge_cmap和edge_vmin,edge_vmax引數。

style:string
   邊線樣式(預設='solid')(實線|虛線|點線,dashdot)

alpha:float
   邊緣透明度(預設值= 1.0)

edge_ cmap:Matplotlib色彩對映
   用於對映邊緣強度的色彩對映(預設值=無)

edge_vmin,edge_vmax:float
   邊緣色圖縮放的最小值和最大值(預設值=無)

ax:Matplotlib Axes物件,可選
   在指定的Matplotlib軸中繪製圖形。

arrows:bool,optional(default = True)
   對於有向圖,如果為真,則繪製箭頭。

label:圖例的標籤
def draw_networkx_nodes(G, pos,
                        nodelist=None,
                        node_size=300,
                        node_color='r',
                        node_shape='o',
                        alpha=1.0,
                        cmap=None,
                        vmin=None,
                        vmax=None,
                        ax=None,
                        linewidths=None,
                        label=None,
                        **kwds):

G:圖表
一個networkx圖

pos:dictionary
   將節點作為鍵和位置作為值的字典。
   位置應該是長度為2的序列。

ax:Matplotlib Axes物件,可選
   在指定的Matplotlib軸中繪製圖形。

nodelist:list,可選
   只繪製指定的節點(預設G.nodes())

node_size:標量或陣列
   節點大小(預設值= 300)。如果指定了陣列,它必須是
   與點頭長度相同。

node_color:顏色字串或浮點陣列
   節點顏色。可以是單顏色格式字串(default ='r'),
   或者具有與點頭相同長度的顏色序列。
   如果指定了數值,它們將被對映到
   顏色使用cmap和vmin,vmax引數。看到
   matplotlib.scatter更多詳細資訊。

node_shape:string
   節點的形狀。規格為matplotlib.scatter
   標記,'so ^> v <dph8'(預設='o')之一。

alpha:float
   節點透明度(預設值= 1.0)

cmap:Matplotlib色圖
   色彩對映節點的強度(預設=無)

vmin,vmax:float
   節點色彩對映縮放的最小值和最大值(預設值=無)

線寬:[無|標量|序列]
   符號邊框的線寬(預設值= 1.0)

label:[無|串]
   圖例的標籤

然後基本上所有networkx的超酷精美圖的原始碼你都能快速弄懂了。
http://networkx.github.io/ 網路圖案例原始碼
這裡寫圖片描述