绘制动画网络图的工具是什么

时间:2018-05-16 16:41:45

标签: python graph networkx

我想获得复杂图形上随机游走概率分布的动画。我目前使用Python和NetworkX来操作图形和评估步行的动态。

我的目标是拥有一个动画(例如,一个GIF文件),其中图形的每个节点的大小与其度(或其他拓扑属性)成比例,并且颜色与标量属性(概率分布)成比例。 节点的大小和位置在时间上保持固定,但颜色会发生变化。

目前,我可以使用Gephi在特定时刻绘制具有所需属性的图形,但我想知道如何制作动画,或者如何自动化每个时刻生成一个图像。

有人可以指出一些已经完成类似事情的参考吗?我还可以使用除Gephi之外的其他可视化工具。实际上,理想情况下,我会在Python中使用我的所有工作流而不需要使用外部程序。

1 个答案:

答案 0 :(得分:3)

在matplotlib中使用FuncAnimation相当简单:

import numpy as np
import matplotlib.pyplot as plt; plt.close('all')
import networkx as nx
from matplotlib.animation import FuncAnimation

def animate_nodes(G, node_colors, pos=None, *args, **kwargs):

    # define graph layout if None given
    if pos is None:
        pos = nx.spring_layout(G)

    # draw graph
    nodes = nx.draw_networkx_nodes(G, pos, *args, **kwargs)
    edges = nx.draw_networkx_edges(G, pos, *args, **kwargs)
    plt.axis('off')

    def update(ii):
        # nodes are just markers returned by plt.scatter;
        # node color can hence be changed in the same way like marker colors
        nodes.set_array(node_colors[ii])
        return nodes,

    fig = plt.gcf()
    animation = FuncAnimation(fig, update, interval=50, frames=len(node_colors), blit=True)
    return animation

total_nodes = 10
graph = nx.complete_graph(total_nodes)
time_steps = 20
node_colors = np.random.randint(0, 100, size=(time_steps, total_nodes))

animation = animate_nodes(graph, node_colors)
animation.save('test.gif', writer='imagemagick', savefig_kwargs={'facecolor':'white'}, fps=0.5)

enter image description here