python实现从大图中筛选出小的子图并导成json文件

简述：

在做数据可视化时会遇到这样一个问题：比如说在浏览器渲染出一个力学关系图，如果结点太多的话就密密麻麻，层层叠叠难以观察，所以往往只需要渲染出关键的结点子图即可，那么有一个解决思路就是用python做数据处理筛选出子图并导成前端需要的格式文件，我写的这个python文件就是模拟实现这个功能。

实现思路：

1、引入需要的数据文件和定义好要生成的文件：

这个python文件是对图数据文件进行操作，使用的数据文件有边文件（里面就包含了结点id），和四个结点指标数据文件。

 

2、定义好要用到的关键变量：

 

其中TopNum很重要，它表示某一指标要取前5的结点。

 

3、将各个指标值保存在相应的数组中，并对数组排序，如果值大的比较重要就倒序。这个操作完成后取这个指标值top5对应的结点id。

 

 

4、获取第一层结点后便是借鉴广度优先搜索的思想依次获取第二层、第三层和第四层。中间为了要做数组过滤去重。

 

 

5、将筛选出的子图id和指标值保存到一个文件里,将子图id排序并映射成从0开始的连续id值。

 

文件保存结果：

 

6、借助映射文件获得结点映射过后的边文件：

 

 

数据文件：

 

 

7、最后根据以上文件生成子图json格式文件：

 

下面是代码地址：

Github地址