Python基于network模块制作电影人物关系图


Posted in Python onJune 19, 2020

在我们生活的世界中,每一个人以及每一个事物相互之间都存在着关系,有直接关系,也有间接关系,最终会形成一个无形的大的关系网。network模块是一个用python语言开发的图论和复杂网络建模工具,模块内置了常用的图与复杂网络分析算法。

network模块有四种图:Graph、DiGraph、MultiGraph、MultiDigraph,分别为无多重边无向图、无多重边有向图、有多重边无向图、有多重边有向图。其中Graph是用点和线来刻画离散事物集合中,每对事物间以某种方式相联系的数学模型。

下面我们来分析《复仇者联盟4》人物关系:

import pandas as pd
#导入绘图模块
import networkx as nx
import matplotlib.pyplot as plt
# 读取文件
aa =r'F:\\python入门\\python编程锦囊\\Code(实例源码及使用说明)\\Code(实例源码及使用说明)\\Code(实例源码及使用说明)\\09\\data\\fl4.xls'
df = pd.DataFrame(pd.read_excel(aa))
#去除重复项,并转换成列表
df1=df['label1'].drop_duplicates().values.tolist()
df2=df[['label1','label2','weight']]
#设置画布大小
plt.figure(figsize=(6, 5))
#颜色数据
colors = df['color'].drop_duplicates().values.tolist()
#G:图表,一个networkx图
G = nx.Graph()
# 添加边
for i in df2.index:
  G.add_edge(df2.label1[i], df2.label2[i], weight=df2.weight[i])
# 定义两个边,并给边赋予权重,其中u是起点,v是终点,d是权重
edge1 = [(u, v) for (u, v, d) in G.edges(data=True) if (d['weight'] >=1)]
edge2 = [(u, v) for (u, v, d) in G.edges(data=True) if (d['weight'] >=15)]
# 图的布局
# 节点在一个圆环上均匀分布
pos = nx.circular_layout(G)
#用Fruchterman-Reingold算法排列节点
#pos=nx.spring_layout(G)
#节点随机分布
#pos=nx.spring_layout(G)
# 点
#node_size指定节点的尺寸大小,默认值为300
#node_color指定节点的颜色,默认值为红色
#node_shape节点的形状,默认值为圆形,用o表示
nx.draw_networkx_nodes(G, pos, alpha=1, node_size=200,node_color=colors,node_shape='o')
#nx.draw_networkx_nodes(G, pos, alpha=1, node_size=300,node_color=colors,node_shape='p')
# 边
#pos:字典类型,节点作为键、位置作为值。位置是长度为2的序列
#edgelist:边缘元组的集合,只绘制指定的边,默认值为G.edges()
#width边的宽度,默认值为1.0
#alpha透明度,默认值为1.0(不透明),0为完全透明
#edge_color边的颜色,默认值为黑色
#style边的样式,默认值为实线。
nx.draw_networkx_edges(G, pos, edgelist=edge1,width=1, alpha=0.3, edge_color='g', style='dashed')
nx.draw_networkx_edges(G, pos, edgelist=edge2, width=1.5, alpha=0.5, edge_color='red')
# 标签
#font_size节点标签字体大小,默认值为12
nx.draw_networkx_labels(G, pos, font_size=9)
# 生成结果
plt.axis('off')
plt.title('《复仇者联盟4》人物关系图')
plt.rcParams['font.size'] = 10
plt.rcParams['font.sans-serif']=['SimHei'] #解决中文乱码
plt.show()

结果:

Python基于network模块制作电影人物关系图

使用力引导算法Fruchterman-Reingold排列点画图,可以大大减少边的交叉,只需要改两行代码即可:

#用Fruchterman-Reingold算法排列节点
pos=nx.spring_layout(G)
# 点
nx.draw_networkx_nodes(G, pos, alpha=1, node_size=300,node_color=colors,node_shape='p')

结果:

Python基于network模块制作电影人物关系图

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用WMI检测windows系统信息、硬盘信息、网卡信息的方法
May 15 Python
将Python的Django框架与认证系统整合的方法
Jul 24 Python
简单实现python爬虫功能
Dec 31 Python
Python编程求质数实例代码
Jan 31 Python
python3.6.3转化为win-exe文件发布的方法
Oct 31 Python
对Python定时任务的启动和停止方法详解
Feb 19 Python
python跳出双层for循环的解决方法
Jun 24 Python
对Python _取log的几种方式小结
Jul 25 Python
Python多继承以及MRO顺序的使用
Nov 11 Python
python开发实例之python使用Websocket库开发简单聊天工具实例详解(python+Websocket+JS)
Mar 18 Python
python自动下载图片的方法示例
Mar 25 Python
Python用户自定义异常的实现
Dec 25 Python
keras中的History对象用法
Jun 19 #Python
python中rc1什么意思
Jun 19 #Python
解决keras backend 越跑越慢问题
Jun 18 #Python
Python基于Twilio及腾讯云实现国际国内短信接口
Jun 18 #Python
详解Windows下PyCharm安装Numpy包及无法安装问题解决方案
Jun 18 #Python
解决Keras中Embedding层masking与Concatenate层不可调和的问题
Jun 18 #Python
Win10下用Anaconda安装TensorFlow(图文教程)
Jun 18 #Python
You might like
PHP新手上路(六)
2006/10/09 PHP
PHP中file_exists()判断中文文件名无效的解决方法
2014/11/12 PHP
PHP保存带BOM文件的方法
2015/02/12 PHP
使用xampp搭建运行php虚拟主机的详细步骤
2015/10/21 PHP
javascript读取xml
2006/11/04 Javascript
验证用户是否修改过页面的数据的实现方法
2008/09/26 Javascript
提高网站性能之 如何对待JavaScript
2009/10/31 Javascript
JQuery AJAX 中文乱码问题解决
2013/06/05 Javascript
js和php如何获取当前url的内容
2013/09/22 Javascript
JavaScript脚本库编写的方法
2015/12/09 Javascript
JavaScript中Function函数与Object对象的关系
2015/12/17 Javascript
AngularJS 让人爱不释手的八种功能
2016/03/23 Javascript
javascript常见数字进制转换实例分析
2016/04/21 Javascript
BootStrap glyphicons 字体图标实现方法
2016/05/01 Javascript
jQuery的ajax下载blob文件
2016/07/21 Javascript
利用jQuery实现打字机字幕效果实例代码
2016/09/02 Javascript
快速实现jQuery多级菜单效果
2017/02/01 Javascript
跟老齐学Python之变量和参数
2014/10/10 Python
Python中暂存上传图片的方法
2015/02/18 Python
Python多线程下载文件的方法
2015/07/10 Python
Python编码类型转换方法详解
2016/07/01 Python
分分钟入门python语言
2018/03/20 Python
Python基于SMTP协议实现发送邮件功能详解
2018/08/14 Python
python 定时器,实现每天凌晨3点执行的方法
2019/02/20 Python
Python 运行.py文件和交互式运行代码的区别详解
2019/07/02 Python
python设置随机种子实例讲解
2019/09/12 Python
Python将二维列表list的数据输出(TXT,Excel)
2020/04/23 Python
Pythonic版二分查找实现过程原理解析
2020/08/11 Python
HTML5之SVG 2D入门11—用户交互性(动画)介绍及应用
2013/01/30 HTML / CSS
英国航空官网:British Airways
2016/09/11 全球购物
应聘编辑职位自荐信范文
2014/01/05 职场文书
陪护人员误工证明
2015/06/24 职场文书
大学生先进个人主要事迹材料
2015/11/04 职场文书
学校趣味运动会开幕词
2016/03/04 职场文书
银行求职信范文
2019/05/13 职场文书
导游词之青岛太清宫
2019/12/13 职场文书