楼赛 第 28 期 Python 通关挑战

背景介绍

新项目成立不久，公司安排项目部的 300 名员工分批前往各地调研，一共 10 批。因为人名单中有员工的手机号，公司决定按手机尾号进行分组。

挑战步骤

下面是300位员工的人名单，请下载到 /home/project 目录下进行实验。

wget https://labfile.oss.aliyuncs.com/courses/4117/人名单.csv

注：该名单使用 faker 库随机生成，非真实数据。

截图如下：

在 /home/project 目录下新建 group.py 文件实现自动读取文件，并自动创建 分组 文件夹，将分组后的名单放在 分组 文件夹下面，分组文件和源文件格式一样，都是 .csv 格式，都只有两列。

挑战要求

• 题目需使用 Python 3 完成。
• python 代码必须写在 /home/project/group.py 文件中。
• 代码需要实现在脚本所在目录下自动创建 分组 文件夹，所有生成的csv文件都放在 分组 文件夹下。
• 新生成的分组文件必须有表头。
• csv 文件命名规则为 尾号0.csv、尾号1.csv 等。
• 最终实现效果如下（图 2 为部分截图）：

import os
f = open('人名单.csv','r')
if not os.path.exists('分组'):
    os.mkdir('分组')
head = f.readline()
# print(head)
lines = f.readlines()
# print(lines[0])
for i in range(0,10):
    f = open('分组/尾号{}.csv'.format(i), 'w+')
    f.write(head)
    f.close
for line in lines:
    # print(line[-2])
    f = open('分组/尾号{}.csv'.format(line[-2]),'a+')
    f.write(line)
    f.close()