接上一篇:/seoyundu/article/details/81543656
代码中:对csv文件读取,并利用pandas库处理,统计出好友信息。
pandas库安装:pip install pandas
测试环境为python3
实现“获取微信好友列表信息”(二)进行导出微信好友到csv数据读取与处理”完整代码如下。
import pandas as pddf = pd.read_csv('friend2_v2.csv', encoding="ISO-8859-1") # 奇怪的编码def city():"""微信朋友圈的城市"""address = df['city'].value_counts() # cityprint(address)def gender():"""微信朋友的性别比例1:男 2:女 3:未知"""gender = df['male'].value_counts() # maleprint(gender)def star():"""星标好友1:星标 0:非星标"""star = df['star'].value_counts() # starprint(star)def remark():remark = df['remark'] # remark MemberCountname = df['name'] # namestar = df['star']print('type(remark)', type(remark))remarkCount = 0maleCount = 0femaleCount = 0for i in range(1, len(remark)):if star[i] != 'nan' or star[i] != '':if str(remark[i]).strip() == str(name[i]).strip() or remark[i] == 'noremark': # remark[i] == 'noremark'remarkCount = remarkCount + 1 # 不认识的人,没有备注else:if judgeGender(i) == 'male':maleCount = maleCount + 1elif judgeGender(i) == 'female':femaleCount = femaleCount + 1print('微信总朋友人数:', str(len(remark)), '\n未标记的人数:', remarkCount, '\n')print('预计认识的总人数:', str(len(remark) - remarkCount), '\n')print('认识的人中汉子人数:', maleCount, '妹子人数:', femaleCount)def judgeGender(index):"""判断传入的某个位置的用户的性别参数:int行返回结果:字符串"""gender = df['male']if gender[index] == 1: # 不要‘1’这样return 'male'elif gender[index] == 2:return 'female'else:return 'unknown'if __name__ == '__main__':remark()
两篇文章完成了微信好友列表提取与导出微信好友列表