文件内容处理
# step1 : 按照excel 处理库
# pip install xlwt
import xlwt
# 使用readline()读文件
f = open("adult.data",encoding='utf-8')
excel = xlwt.Workbook(encoding='utf-8') #创建Workbook,相当于创建Excel
# 创建sheet,Sheet1为表的名字,cell_overwrite_ok为是否覆盖单元格
sheet1 = excel.add_sheet('Sheet1', cell_overwrite_ok=True)
#向表中添加数据
sheet1.write(0, 0, 'income') #第一行第0列
#sheet1.write(0, 1, '中文名字')
#sheet1.write(1, 1, '海伦')
arr =0
#while True:
#for i in range(100):
while True:
line = f.readline()
arr = arr +1
if line:
#print (line)
x = line.find("<=50K")
if x != -1 :
#print('arr = ',arr,'value= ',0)
sheet1.write(arr, 0, '0') #第arr行第0列
else :
#print('arr = ',arr,'value= ',1)
sheet1.write(arr, 0, '1') #第arr行第0列
else:
break
print('处理完毕')
# excel文件保存
excel.save('Excel_test.csv')
#文件保存
f.close()