python豆瓣用户关注网络

chobit

初学半个月，写了这样一个脚本，用于获取用户关注列表，组成网络（我只能说我是跟老师做社会网络，数据挖掘的）。发上来，给大家看看，希望大家能给点意见，初学，肯定有很多需要修改的地方。

1. # python 3.4 gui
   
2. import urllib.request
   
3. import json
   
4. import time
   
5. # import easygui as g
   
6. import sys
   
7. 
   
8. '''
   
9. 函数定义
   
10. '''
    
11. def user_follow(user_name):
    
12.         api_key = '回复不可见' # 我申请的key
    
13.         start_index = '1'
    
14.         max_results = '50'#目前最大
    
15. 
    
16.         '''
    
17.         函数定义
    
18.         '''
    
19.         def connectfunction():
    
20.                 succeedornot = 0
    
21.                 while succeedornot == 0:
    
22.                         try:
    
23.                                 url_conn = urllib.request.urlopen('http://api.douban.com/people/' + user_name + '/contacts?apikey=' + api_key + '&alt=json&start-index=' + str(start_index) + '&max-results=' + str(max_results)) # 打开url
    
24.                                 succeedornot = 1
    
25.                         except (TimeoutError,urllib.error.URLError):
    
26.                                 print('网络不稳定，正在重试...')
    
27.                                 succeedornot = 0
    
28.                 return url_conn
    
29.                
    
30.         File1 = connectfunction()
    
31.         doubanHTML = File1.read().decode('utf-8') # 读入打开的url
    
32.         doubanJSON = json.JSONDecoder().decode(doubanHTML) # 创建json
    
33. 
    
34.         # 信息截取
    
35.         # openSearch:itemsPerPage
    
36.         itemsPerPage = doubanJSON['openSearch:itemsPerPage']['$t'] # 每次获取的关注人数
    
37.        
    
38.         # author
    
39.         author = doubanJSON['author'] # 研究对象
    
40.        
    
41.         #         name
    
42.         author_name = author['name']['$t'] # 用户名称
    
43. 
    
44.         # openSearch:totalResults
    
45.         totalResults = doubanJSON['openSearch:totalResults']['$t'] # 关注人数
    
46.        
    
47.         # 打印信息
    
48.         print('关注人数：', totalResults)
    
49.         print('每次获取的关注人数：', itemsPerPage)
    
50.         print('对象名称：', author_name)
    
51. 
    
52.         # file_name = 'E:\\' + author_name + '_follow_file.txt'
    
53.         file_name = 'E:\\follow_file.txt'
    
54.         follow_list_pair = []
    
55.         follow_list_solo = []
    
56.         follow_file = open(file_name,'a',encoding='utf-8')# 以写入模式打开，如果文件存在，则在末尾追加写入
    
57. 
    
58.         '''
    
59.         函数定义
    
60.         '''
    
61.         def milk_run(contact_list,contact_index):
    
62.                 #        db:uid
    
63.                 uid = contact_list[contact_index]['db:uid']['$t'] # 关注对象uid
    
64.                
    
65.                 #        db:title
    
66.                 title = contact_list[contact_index]['title']['$t'] # 关注对象名称
    
67.                
    
68.                 return (title,uid)
    
69. 
    
70.         '''
    
71.         循环开始
    
72.         '''
    
73.         while int(start_index) <= int(totalResults): # 循环分段获取
    
74.                 if int(start_index) + int(max_results) - 1 <= int(totalResults):
    
75.                         end_index = int(start_index) + int(max_results) - 1 # 确定范围
    
76.                         subtract = int(max_results)
    
77.                 else:
    
78.                         end_index = int(totalResults)
    
79.                         subtract = end_index - int(start_index) + 1
    
80.                 print(start_index,'-',end_index)
    
81.                
    
82.                 File1 = connectfunction()
    
83.                 doubanHTML = File1.read().decode('utf-8') # 读入打开的url
    
84.                 doubanJSON = json.JSONDecoder().decode(doubanHTML) # 创建json
    
85. 
    
86.                 # entry
    
87.                 entry = doubanJSON['entry'] # 关注对象列表
    
88. 
    
89.                 for index in range(subtract):
    
90.                         follow_list_pair.append(user_name + '        ' + milk_run(entry,index)[1] + '\n')# 存入列表
    
91.                         follow_list_solo.append(milk_run(entry,index)[1])# 存入列表
    
92.                         print('第' + str(int(start_index) + index) + '个关注：',milk_run(entry,index),'，已保存')
    
93.                
    
94.                 start_index = int(start_index) + int(max_results) # 步进
    
95.                 str(start_index)
    
96. 
    
97.         follow_file.writelines(follow_list_pair)
    
98.         follow_file.close()
    
99.        
    
100.         return follow_list_solo
     
101. 
     
102. # user_input = g.enterbox(msg='请输入用户id或uid:', title='查询用户关注网络', default='ahbei', strip=True, image=None, root=None)       
     
103. user_input = input('请输入用户名称:')
     
104. done_user = []
     
105. done_user.append(user_input)
     
106. user_input_follow = user_follow(user_input)
     
107. 
     
108. for followers in user_input_follow:# 二层
     
109.         if str(followers) not in done_user:
     
110.                 done_user.append(str(followers))
     
111.                 user_follow_follow = user_follow(str(followers))
     
112.                 # for followers_2 in user_follow_follow:# 三层
     
113.                         # if str(followers_2) not in done_user:
     
114.                                 # done_user.append(str(followers_2))
     
115.                                 # user_follow_follow_follow = user_follow(str(followers_2))
     
116.                                 # time.sleep(2) # 延迟2s
     

复制代码

后期会使用igraph进行网络分析（这个我以前用R语言玩过）

游客，如果您要查看本帖隐藏内容请回复

douban_gui.zip (1.66 KB, 下载次数: 7, 售价: 1 鱼币)

2014-6-1 09:55 上传

点击文件名下载附件
阅读权限: 5

zach_chen

谢楼主分享

毛桃子

楼主，你好，我最近刚接触R语言的igraph但是很多不懂的地方，能不能请教你一下啊？

xiaoshuai007

学习学习