python用三位数给百家姓排序
发布网友
发布时间：2022-04-25 22:34
共1个回答
热心网友
时间：2022-06-18 05:42
import sys
import os
import csv
import codecs
import numpy as np
import tensorflow as tf
from datetime import datetime
file_dir = './综合作业A'
'''
for root, dirs, files in os.walk(file_dir):
#print(root) #当前目录路径
print(dirs) #当前路径下所有子目录
#print(files) #当前路径下所有非目录子文件
'''
import unicodecsv as ucsv
import natsort
from xpinyin import Pinyin
f = open('homework.csv','wb')
#f.write(codecs.BOM_UTF8)
#f.write(u'\ufeff'.encode('utf8'))
csv_writer = ucsv.writer(f, encoding = 'gbk')
inform = os.listdir(file_dir)
#natsort.natsorted(inform)
def my_function(lis): #输入一个名字的列表
pin=Pinyin()
result=[]
for item in lis:
result.append((pin.get_pinyin(item.split('(')[0]),item))
result.sort()
for i in range(len(result)):
result[i]=result[i][1]
#result=' '.join(result) #将排好序的结果使用空格连接，方便输出
return result #输出结果
inform = my_function(inform)
for var in inform:
print(var)
name = var.split('(')[0]
no = var.split('(')[1][0:-1]
print(name)
print(no)
csv_writer.writerow([no, name])
#csv_writer.writerow([])
f.close()
