python项目：获取微信好友信息（二）csv数据读取与处理

本文将记录对知乎专栏中的一篇小项目，自己尝试中所遇到的问题。

知乎链接：https://zhuanlan.zhihu.com/p/21967333

代码中：对csv文件读取，并利用pandas库处理，统计出好友信息。

pandas库安装：pip  install pandas

测试环境为python3

完整代码如下。

import pandas as pd

df = pd.read_csv('friend2_v2.csv', encoding="ISO-8859-1")  # 奇怪的编码

def city():
    """微信朋友圈的城市"""
    address = df['city'].value_counts()  # city
    print(address)


def gender():
    """微信朋友的性别比例
        1:男  2：女   3：未知
    """
    gender = df['male'].value_counts()  # male
    print(gender)


def star():
    """星标好友
        1:星标   0：非星标
    """
    star = df['star'].value_counts()  # star
    print(star)


def remark():
    remark = df['remark']  # remark MemberCount
    name = df['name']  # name
    star = df['star']
    print('type(remark)', type(remark))

    remarkCount = 0
    maleCount = 0
    femaleCount = 0
    for 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 + 1
                elif judgeGender(i) == 'female':
                    femaleCount = femaleCount + 1
    print('微信总朋友人数：', 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()

在尝试过程中，问题主要会出现在编码问题上，详见： http://blog.csdn.net/qwerty200696/article/details/53016243