Python统计字数并将结果输入到TXT文件中

import  string
import pandas as pd

def str_count(str):
    '''找出字符串中的中英文、空格、数字、标点符号个数'''
    count_en = count_dg = count_sp = count_zh = count_pu = 0


    for s in str:
        data = open("WORD_COUNT_result.txt", 'a')
        # 英文
        if s in string.ascii_letters:
            count_en += 1
        # 数字
        elif s.isdigit():
            count_dg += 1
        # 空格
        elif s.isspace():
            count_sp += 1
        # 中文
        elif s.isalpha():
            count_zh += 1
        # 特殊字符
        else:
            count_pu += 1
    # print('英文字符：', count_en)
    # print('数字：', count_dg)
    # print('空格：', count_sp)
    print('中文：', count_zh)
    # print('特殊字符：', count_pu)
    print(count_zh, file=data)
    data.close()
data = pd.read_csv('renrendai_data_ceshi.csv',encoding='GBK')
data = pd.DataFrame(data)
data['中文字数'] = '' #新加一列用于存放结果
data2 = data['借款描述']
str = data2
for i in range(len(str)):
    str_count(str[i])