轉載自http://blog.csdn.net/rtygbwwwerr/article/details/50937643

代碼是別人寫的，研究了下詳細的用法，如下：

makeNN.py

?

import os import sys import platformfrom string import Templatedef makeNode(index,start=0,shape='circle',num=3,height='.5'):code = Template('''node [shape=$shape,height=$height];\n$nodes;\n''')node_set=[];nodes=''for i in range(num):n = 'a%d' %index + '%d' %(start+i)node_set.append(n)nodes = nodes + (n + ' ')str = code.substitute(index=index, shape=shape, nodes=nodes,height=height)return node_set,strdef makeLayer(index,color='red2',node_num=3, maxNum=2,name=''):code = Template('''subgraph cluster_$index {\n color=white;\nnode [style=solid,color=$color];\n$nodesB$nodesM$nodesElabel = "layer $name($node_num)";\n}\n''')nodesB=''nodesM=''nodesE=''node_set=[];if node_num > maxNum:headNum = maxNum // 2bset,nodesB = makeNode(index=index,num=headNum)_,nodesM = makeNode(index=index,start=headNum,shape='point',height='.02')eset,nodesE = makeNode(index=index,start=(node_num-headNum),num=headNum)node_set = (bset+eset)else:nodes,nodesM = makeNode(index=index,num=node_num)node_set = nodesif name=='':name = indexstr = code.substitute(index=index,name=name, color=color, nodesB=nodesB,nodesM=nodesM,nodesE=nodesE,node_num=node_num)return node_set,strdef makeLines(connects,node_sets):str=''for c in connects:for i1 in node_sets[c[0]]:for i2 in node_sets[c[1]]:str = str + (i1 + ' -> ' + i2) + '\n'return str;def makeFrame(args):code = Template('''digraph G {\n rankdir=LR\nsplines=line\nnodesep=.05;\nnode [label=""];\n$layers\n$lines\n} ''')node_sets=[]layer_num = args["layers_num"]layers_cfg = args['layers_cfg']connects = args['connects']visual_num = args['visual_num']layers = '';for i in range(layer_num):node_set,str = makeLayer(i,layers_cfg[i][2],layers_cfg[i][1],visual_num,layers_cfg[i][0])node_sets.append(node_set)layers = layers + strlines = makeLines(connects,node_sets)return code.substitute(layers=layers,lines=lines)def saveFile(str, path): print (path)f = open(path,'w') f.write(str)f.close()def isWindowsSystem():return 'Windows' in platform.system() def makeConfig():args = sys.argv[1:]if len(args) == 0:args = makeDefConfig()else: fp = open(args[0],'r')args = dict()line = fp.readline();while line:if line.index('#') > 1:item = line.split(':')args.update({item[0]:item[1]})line = fp.readline()return argsdef makeDefConfig():args = dict()#pathpath = sys.argv[0]if isWindowsSystem():last = path.rindex('\\')else:last = path.rindex('/')path = path[0:last+1]args.update({'input':path + 'nn.gv'})args.update({'output':path + 'nn.png'})#layer numberargs.update({'visual_num':10})#layers config:(name,nodes number,color)args.update({'layers_cfg':(('input',10,'blue4'),('h1',7,'red2'),('h2',5,'red2'),('h3',4,'red2'),('out',2,'seagreen2'))})layers = args['layers_cfg']args.update({'layers_num':len(layers)})#connects:layer_i->lay_jargs.update({'connects':([0,1],[1,2],[2,3],[3,4])})return argsif __name__ == "__main__":args = makeConfig()str = makeFrame(args)saveFile(str, args['input'])cmd = 'dot ' + args['input'] + ' -Tpng -o ' + args['output'] os.system(cmd)

?

?

?

?

?

nn.py

?

?

digraph G {rankdir=LRsplines=linenodesep=.05;node [label=""];subgraph cluster_0 {color=white;node [style=solid,color=blue4];node [shape=circle,height=.5];a00 a01 a02 a03 a04 a05 a06 a07 a08 a09 ;label = "layer input(10)";}subgraph cluster_1 {color=white;node [style=solid,color=red2];node [shape=circle,height=.5];a10 a11 a12 a13 a14 a15 a16 ;label = "layer h1(7)";}subgraph cluster_2 {color=white;node [style=solid,color=red2];node [shape=circle,height=.5];a20 a21 a22 a23 a24 ;label = "layer h2(5)";}subgraph cluster_3 {color=white;node [style=solid,color=red2];node [shape=circle,height=.5];a30 a31 a32 a33 ;label = "layer h3(4)";}subgraph cluster_4 {color=white;node [style=solid,color=seagreen2];node [shape=circle,height=.5];a40 a41 ;label = "layer out(2)";}a00 -> a10 a00 -> a11 a00 -> a12 a00 -> a13 a00 -> a14 a00 -> a15 a00 -> a16 a01 -> a10 a01 -> a11 a01 -> a12 a01 -> a13 a01 -> a14 a01 -> a15 a01 -> a16 a02 -> a10 a02 -> a11 a02 -> a12 a02 -> a13 a02 -> a14 a02 -> a15 a02 -> a16 a03 -> a10 a03 -> a11 a03 -> a12 a03 -> a13 a03 -> a14 a03 -> a15 a03 -> a16 a04 -> a10 a04 -> a11 a04 -> a12 a04 -> a13 a04 -> a14 a04 -> a15 a04 -> a16 a05 -> a10 a05 -> a11 a05 -> a12 a05 -> a13 a05 -> a14 a05 -> a15 a05 -> a16 a06 -> a10 a06 -> a11 a06 -> a12 a06 -> a13 a06 -> a14 a06 -> a15 a06 -> a16 a07 -> a10 a07 -> a11 a07 -> a12 a07 -> a13 a07 -> a14 a07 -> a15 a07 -> a16 a08 -> a10 a08 -> a11 a08 -> a12 a08 -> a13 a08 -> a14 a08 -> a15 a08 -> a16 a09 -> a10 a09 -> a11 a09 -> a12 a09 -> a13 a09 -> a14 a09 -> a15 a09 -> a16 a10 -> a20 a10 -> a21 a10 -> a22 a10 -> a23 a10 -> a24 a11 -> a20 a11 -> a21 a11 -> a22 a11 -> a23 a11 -> a24 a12 -> a20 a12 -> a21 a12 -> a22 a12 -> a23 a12 -> a24 a13 -> a20 a13 -> a21 a13 -> a22 a13 -> a23 a13 -> a24 a14 -> a20 a14 -> a21 a14 -> a22 a14 -> a23 a14 -> a24 a15 -> a20 a15 -> a21 a15 -> a22 a15 -> a23 a15 -> a24 a16 -> a20 a16 -> a21 a16 -> a22 a16 -> a23 a16 -> a24 a20 -> a30 a20 -> a31 a20 -> a32 a20 -> a33 a21 -> a30 a21 -> a31 a21 -> a32 a21 -> a33 a22 -> a30 a22 -> a31 a22 -> a32 a22 -> a33 a23 -> a30 a23 -> a31 a23 -> a32 a23 -> a33 a24 -> a30 a24 -> a31 a24 -> a32 a24 -> a33 a30 -> a40 a30 -> a41 a31 -> a40 a31 -> a41 a32 -> a40 a32 -> a41 a33 -> a40 a33 -> a41}

?

?

nna.gv

?

digraph G {rankdir=LRsplines=linenodesep=.05;node [label=""];subgraph cluster_0 {color=white;node [style=solid,color=blue4, shape=circle];x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 x10 x11 x12 x13 x14 x15 x16 x17 x18 x19 x20 x21 x22 x23 x24 x25 x26 x27 x28 x29 x30 x31 x32 x33 x34 x35 x36 x37 x38 x39 x40 x41 x42 x43 x44 x45 x46 x47 x48 x49 x50 x51 x52 x53 x54 x55 x56 x57 x58 x59 x60 x61 x62 x63 x64 x65 x66 x67 x68 x69 x70 x71 x72 x73 x74 x75 x76 x77 x78 x79 x80 x81 x82 x83 x84 x85 x86 x87 x88 x89 x90 x91 x92 x93 x94 x95 x96 x97 x98 x99 ;label = "layer 0";}subgraph cluster_1 {color=white;node [style=solid,color=red2, shape=circle];a10 a11 a12 a13 a14 a15 a16 a17 a18 a19 a110 a111 a112 a113 a114 a115 a116 a117 a118 a119 a120 a121 a122 a123 a124 a125 a126 a127 a128 a129 a130 a131 a132 a133 a134 a135 a136 a137 a138 a139 a140 a141 a142 a143 a144 a145 a146 a147 a148 a149 a150 a151 a152 a153 a154 a155 a156 a157 a158 a159 a160 a161 a162 a163 a164 a165 a166 a167 a168 a169 a170 a171 a172 a173 a174 a175 a176 a177 a178 a179 a180 a181 a182 a183 a184 a185 a186 a187 a188 a189 a190 a191 a192 a193 a194 a195 a196 a197 a198 a199 a1100 a1101 a1102 a1103 a1104 a1105 a1106 a1107 a1108 a1109 a1110 a1111 a1112 a1113 a1114 a1115 a1116 a1117 a1118 a1119 ;label = "layer 1";}subgraph cluster_2 {color=white;node [style=solid,color=red2, shape=circle];a20 a21 a22 a23 a24 a25 a26 a27 a28 a29 a210 a211 a212 a213 a214 a215 a216 a217 a218 a219 a220 a221 a222 a223 a224 a225 a226 a227 a228 a229 a230 a231 a232 a233 a234 a235 a236 a237 a238 a239 a240 a241 a242 a243 a244 a245 a246 a247 a248 a249 a250 a251 a252 a253 a254 a255 a256 a257 a258 a259 a260 a261 a262 a263 a264 a265 a266 a267 a268 a269 a270 a271 a272 a273 a274 a275 a276 a277 a278 a279 a280 a281 a282 a283 a284 a285 a286 a287 a288 a289 a290 a291 a292 a293 a294 a295 a296 a297 a298 a299 a2100 a2101 a2102 a2103 a2104 a2105 a2106 a2107 a2108 a2109 a2110 a2111 a2112 a2113 a2114 a2115 a2116 a2117 a2118 a2119 ;label = "layer 2";}subgraph cluster_3 {color=white;node [style=solid,color=red2, shape=circle];a30 a31 a32 a33 a34 a35 a36 a37 a38 a39 a310 a311 a312 a313 a314 a315 a316 a317 a318 a319 a320 a321 a322 a323 a324 a325 a326 a327 a328 a329 a330 a331 a332 a333 a334 a335 a336 a337 a338 a339 a340 a341 a342 a343 a344 a345 a346 a347 a348 a349 a350 a351 a352 a353 a354 a355 a356 a357 a358 a359 a360 a361 a362 a363 a364 a365 a366 a367 a368 a369 a370 a371 a372 a373 a374 a375 a376 a377 a378 a379 a380 a381 a382 a383 a384 a385 a386 a387 a388 a389 a390 a391 a392 a393 a394 a395 a396 a397 a398 a399 a3100 a3101 a3102 a3103 a3104 a3105 a3106 a3107 a3108 a3109 a3110 a3111 a3112 a3113 a3114 a3115 a3116 a3117 a3118 a3119 a3120 a3121 a3122 a3123 a3124 a3125 a3126 a3127 a3128 a3129 a3130 a3131 a3132 a3133 a3134 a3135 a3136 a3137 a3138 a3139 a3140 a3141 a3142 a3143 a3144 a3145 a3146 a3147 a3148 a3149 ;label = "layer 3";}subgraph cluster_4 {color=white;node [style=solid,color=seagreen2, shape=circle];a40 a41 ;label = "layer 4";}x0 -> a10 x0 -> a11 x0 -> a12 x0 -> a13 x0 -> a14 x0 -> a15 x0 -> a16 x0 -> a17 x0 -> a18 x0 -> a19 x0 -> a110 x0 -> a111 x0 -> a112 x0 -> a113 x0 -> a114 x0 -> a115 x0 -> a116 x0 -> a117 x0 -> a118 x0 -> a119 x0 -> a120 x0 -> a121 x0 -> a122 x0 -> a123 x0 -> a124 x0 -> a125 x0 -> a126 x0 -> a127 x0 -> a128 x0 -> a129 x0 -> a130 x0 -> a131 x0 -> a132 x0 -> a133 x0 -> a134 x0 -> a135 x0 -> a136 x0 -> a137 x0 -> a138 x0 -> a139 x0 -> a140 x0 -> a141 x0 -> a142 x0 -> a143 x0 -> a144 x0 -> a145 x0 -> a146 x0 -> a147 x0 -> a148 x0 -> a149 x0 -> a150 x0 -> a151 x0 -> a152 x0 -> a153 x0 -> a154 x0 -> a155 x0 -> a156 x0 -> a157 x0 -> a158 x0 -> a159 x0 -> a160 x0 -> a161 x0 -> a162 x0 -> a163 x0 -> a164 x0 -> a165 x0 -> a166 x0 -> a167 x0 -> a168 x0 -> a169 x0 -> a170 x0 -> a171 x0 -> a172 x0 -> a173 x0 -> a174 x0 -> a175 x0 -> a176 x0 -> a177 x0 -> a178 x0 -> a179 x0 -> a180 x0 -> a181 x0 -> a182 x0 -> a183 x0 -> a184 x0 -> a185 x0 -> a186 x0 -> a187 x0 -> a188 x0 -> a189 x0 -> a190 x0 -> a191 x0 -> a192 x0 -> a193 x0 -> a194 x0 -> a195 x0 -> a196 x0 -> a197 x0 -> a198 x0 -> a199 x0 -> a1100 x0 -> a1101 x0 -> a1102 x0 -> a1103 x0 -> a1104 x0 -> a1105 x0 -> a1106 x0 -> a1107 x0 -> a1108 x0 -> a1109 x0 -> a1110 x0 -> a1111 x0 -> a1112 x0 -> a1113 x0 -> a1114 x0 -> a1115 x0 -> a1116 x0 -> a1117 x0 -> a1118 x0 -> a1119 x1 -> a10 x1 -> a11 x1 -> a12 x1 -> a13 x1 -> a14 x1 -> a15 x1 -> a16 x1 -> a17

config.txt

?

#[*.gv,*.png]=current dir/nn.gv,current dir/nn.png path:, #([visual node number]=3,<([layer name]=layer index,[layer nodes number]=3,[nodes color]='red2')>) node:(3,('input',1,'blue4'),('h2',5,'red2'),('h3',5,'red2'),('h4',5,'red2'),('h5',2,'seagreen2')) #(<[layer1,layer2],...[layeri,layerj]>);connects from layer i to layer j(full connect) connect:([0,1],[0,2],[0,3],[1,4],[2,4],[3,4])

然后把上面四個文件放入一個叫src的文件夾中，在pycharm中運行makeNN.py以后，在src文件夾中會生成一個圖片，如下：

?

?

總結

以上是生活随笔為你收集整理的python绘制神经网络（转载）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。