绘制人人好友关系图代码,Python代码库,德仔网

代码语言
代码分类
【Python】绘制人人好友关系图
作者:沉淀岁月 / 发布于2013/9/13/ 1129
依赖：networkx matplotlib

	#! /bin/env python
	# -*- coding: utf-8 -*-
	
	import urllib
	import urllib2
	import cookielib
	import re
	import cPickle as p
	import networkx as nx
	import matplotlib.pyplot as plt
	
	__author__ = """Reverland (lhtlyy@gmail.com)"""
	
	# Control parameters,EDIT it here
	## Login
	username = 'None'
	password = 'None'
	## Control Graphs, Edit for better graphs as you need
	label_flag = True # Whether shows labels.NOTE: configure your matplotlibrc for Chinese characters.
	remove_isolated = True # Whether remove isolated nodes(less than iso_level connects)
	different_size = True # Nodes for different size, bigger means more shared friends
	iso_level = 10
	node_size = 40 # Default node size
	
	
	def login(username, password):
	 """log in and return uid"""
	 logpage = "http://www.renren.com/ajaxLogin/login"
	 data = {'email': username, 'password': password}
	 login_data = urllib.urlencode(data)
	 cj = cookielib.CookieJar()
	 opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
	 urllib2.install_opener(opener)
	 res = opener.open(logpage, login_data)
	 print "Login now ..."
	 html = res.read()
	 #print html
	
	 # Get uid
	 print "Getting user id of you now"
	 res = urllib2.urlopen("http://www.renren.com/home")
	 html = res.read()
	 # print html
	 uid = re.search("'ruid':'(\\d+)'", html).group(1)
	 # print uid
	 print "Login and got uid successfully"
	 return uid
	
	
	def getfriends(uid):
	 """Get the uid's friends and return the dict with uid as key,name as value."""
	 print "Get %s 's friend list" % str(uid)
	 pagenum = 0
	 dict1 = {}
	 while True:
	 targetpage = "http://friend.renren.com/GetFriendList.do?curpage=" + str(pagenum) + "&id=" + str(uid)
	 res = urllib2.urlopen(targetpage)
	 html = res.read()
	
	 pattern = '<a href="http://www\\.renren\\.com/profile\\.do\\?id=(\\d+)"><img src="[\\S]*" alt="[\\S]*[\\s]\\((.*)\\)" />'
	
	 m = re.findall(pattern, html)
	 #print len(m)
	 if len(m) == 0:
	 break
	 for i in range(0, len(m)):
	 no = m[i][0]
	 uname = m[i][1]
	 #print uname, no
	 dict1[no] = uname
	 pagenum += 1
	 print "Got %s 's friends list successfully." % str(uid)
	 return dict1
	
	
	def getdict(uid):
	 """cache dict of uid in the disk."""
	 try:
	 with open(str(uid) + '.txt', 'r') as f:
	 dict_uid = p.load(f)
	 except:
	 with open(str(uid) + '.txt', 'w') as f:
	 p.dump(getfriends(uid), f)
	 dict_uid = getdict(uid)
	 return dict_uid
	
	
	def getrelations(uid1, uid2):
	 """receive two user id, If they are friends, return 1, otherwise 0."""
	 dict_uid1 = getdict(uid1)
	 if uid2 in dict_uid1:
	 return 1
	 else:
	 return 0
	
	
	def getgraph(username, password):
	 """Get the Graph Object and return it.
	You must specify a Chinese font such as `SimHei` in ~/.matplotlib/matplotlibrc"""
	 uid = login(username, password)
	 dict_root = getdict(uid) # Get root tree
	
	 G = nx.Graph() # Create a Graph object
	 for uid1, uname1 in dict_root.items():
	 # Encode Chinese characters for matplotlib **IMPORTANT**
	 # if you want to draw Chinese labels,
	 uname1 = unicode(uname1, 'utf8')
	 G.add_node(uname1)
	 for uid2, uname2 in dict_root.items():
	 uname2 = unicode(uname2, 'utf8')
	 # Not necessary for networkx
	 if uid2 == uid1:
	 continue
	 if getrelations(uid1, uid2):
	 G.add_edge(uname1, uname2)
	
	 return G
	
	
	def draw_graph(username, password, filename='graph.txt', label_flag=True, remove_isolated=True, different_size=True, iso_level=10, node_size=40):
	 """Reading data from file and draw the graph.If not exists, create the file and re-scratch data from net"""
	 print "Generating graph..."
	 try:
	 with open(filename, 'r') as f:
	 G = p.load(f)
	 except:
	 G = getgraph(username, password)
	 with open(filename, 'w') as f:
	 p.dump(G, f)
	 #nx.draw(G)
	 # Judge whether remove the isolated point from graph
	 if remove_isolated is True:
	 H = nx.empty_graph()
	 for SG in nx.connected_component_subgraphs(G):
	 if SG.number_of_nodes() > iso_level:
	 H = nx.union(SG, H)
	 G = H
	 # Ajust graph for better presentation
	 if different_size is True:
	 L = nx.degree(G)
	 G.dot_size = {}
	 for k, v in L.items():
	 G.dot_size[k] = v
	 node_size = [G.dot_size[v] * 10 for v in G]
	 pos = nx.spring_layout(G, iterations=50)
	 nx.draw_networkx_edges(G, pos, alpha=0.2)
	 nx.draw_networkx_nodes(G, pos, node_size=node_size, node_color='r', alpha=0.3)
	 # Judge whether shows label
	 if label_flag is True:
	 nx.draw_networkx_labels(G, pos, alpha=0.5)
	 #nx.draw_graphviz(G)
	 plt.show()
	
	 return G
	
	if __name__ == "__main__":
	 G = draw_graph(username, password)
试试其它关键字
　好友关系图　
同语言下
可能有用的
沉淀岁月贡献的其它代码(39)