如何保存networkx动画?

2024-04-26 13:58:49 发布

您现在位置:Python中文网/ 问答频道 /正文

我制作了这个函数,它根据循环中生成的给定随机节点标签提供网络动画,但我希望能够录制这个动画,但仍然不知道如何才能做到。 代码如下:

import networkx as nx 
import matplotlib.pyplot as plt
import numpy as np
import random
import time
from matplotlib.animation import FuncAnimation

global ax, fig, G
fig, ax = plt.subplots(figsize=(10,7))
G = nx.DiGraph()

def showNet(n):
    global ax, fig, G

    plt.style.use('fivethirtyeight')
    
    fig.clf()    

    l1, l2, l3, l4, l5, l6, l7 = n[0], n[1], n[2], n[3], n[4], n[5], n[6] 

    edgelist= [(l1,l2),(l2,l3), (l3,l4), (l3,l5), (l4, l1), (l5, l6), (l5, l7)]

    pos = {l1: (10, 60), l2: (10, 40), l3:(80, 40), l4:(140, 60), l5:(200, 20), l6:(250, 40), l7:(250,10)}
    
    nx.draw_networkx_nodes(G, pos=pos, nodelist=list(pos.keys()), node_size=2000, alpha=0.8) 
    nx.draw_networkx_edges(G, pos=pos, edgelist= edgelist , edge_color="gray", arrows=True, arrowsize=10, arrowstyle='wedge') 

    nx.draw_networkx_labels(G, pos=pos, labels=dict(zip(pos.keys(),pos.keys())),  font_color="black") 

    plt.grid(False)

    plt.ion() 
    fig.show() 
    plt.pause(0.0000001)

for i in range(30):
    n = random.sample(range(1, 10), 7)
    showNet(n)

1条回答
网友
1楼 · 发布于 2024-04-26 13:58:49

你可以plt.savefig(f'{n}.png')它们,然后用专门的工具创建一个gif(你可以在互联网上很容易找到)

相关问题 更多 >