参考官网： https://docs.python.org/zh-cn/3.10/library/csv.html
csv 是Python中内置的标准库，不需要下载，可以直接使用。

保存数据到csv文件

基本使用

将数据保存到文件中。

  1. # 导入csv
  2. import csv
  4. def write_csv():
  5.     # 打开文件
  6.     with open('data.csv',encoding='utf8',mode='w') as f:
  7.         # 使用 csv的方式来写入，
  8.         wr =  csv.writer(f)
  9.         # 开始写入数据
  10.         wr.writerow(["手机号","身份证id"])
  12. if __name__ == '__main__':
  13.     write_csv()

执行可以看到对应的效果，生成了一个 data.csv 的文件
image.png

  • writerow() 是csv中提供的方法 可以写入一行数据。 必须传入 一个列表类型的数据。

如果要写入多行数据。 那就使用 for 循环

  1. import csv
  3. def write_csv():
  4.     with open('data.csv',encoding='utf8',mode='w') as f:
  5.         # 使用 csv的方式来写入， 固定用法
  6.         wr =  csv.writer(f)
  7.         # 开始写入数据 往文件中写入数据， 数据必须是 列表格式。写入一行
  8.         wr.writerow(["手机号","身份证id"])
  9.         for i in range(10):
  10.             wr.writerow(["13212231122","111122222334455555"])
  12. if __name__ == '__main__':
  13.     write_csv()

使用for循环 写入10行数据。
但是写入的数据有问题，会多一个空行。
image.png
因为csv 模块在写入文件的时候，会自动添加换行符。在open的时候，指定新行为空的字符串。
image.png

  1. import csv
  3. def write_csv():
  4.     # 默认的csv文件中会空白换行，使用 newline="" 关闭空白换行
  5.     with open('data.csv',encoding='utf8',mode='w',newline="") as f:
  6.         # 使用 csv的方式来写入， 固定用法
  7.         wr =  csv.writer(f)
  8.         # 开始写入数据 往文件中写入数据， 数据必须是 列表格式。写入一行
  9.         wr.writerow(["手机号","身份证id"])
  10.         for i in range(10):
  11.             wr.writerow(["13212231122","111122222334455555"])
  13. if __name__ == '__main__':
  14.     write_csv()

再次运行，可以看到对应的效果。
image.png

读取csv文件的内容

读取data.csv 文件的内容

  1. import csv
  2. def read_csv():
  3.     # 读取权限
  4.     with open(file="data.csv",encoding='utf8',mode='r') as f:
  5.         # reader 读取文件内容 固定用法
  6.         all_rows = csv.reader(f)
  7.         # csv 中有很多行数据，需要一行一行去读
  8.         for line in all_rows:
  9.             print(line)
  11. if __name__ == '__main__':
  12.     # write_csv()
  13.     read_csv()

image.png

解析文件练习

gb2260.csv
上面附件文件中 存放了所有中国地区 身份证号的前6位。请使用python程序将里面的内容打印出来
将上面的文件下载下来，放到 testdata 下， （可以拖拽进去）
image.png
在 common/file_hanlder.py 中定义 一个函数 get_china_code()
通过代码的方式读取 gb2260.csv 文件中的所有地区编码。并随机选择一个。

1. 使用文件绝对路径

文件上—【右键】—【Copy Path】
image.png
选择绝对路径。
image.png

2.读取文件

将复制的绝对文件路径 粘贴进来，字符串前 使用 r

  1. import csv
  2. def get_china_code():
  3.     #  as f 别名 f相当于是 前面open打开的文件  绝对路径字符串 前添加r
  4.     with open(file=r"C:\Users\zengy\PycharmProjects\pythonProject13\testdata\gb2260.csv",
  5.               encoding='utf8',mode='r') as f:
  6.         # 读取完成之后会将所有的内容放在一个列表对象中
  7.         rows = csv.reader(f)
  8.         # 循环列表
  9.         for row in rows:
  10.             print(row)
  12. if __name__ == '__main__':
  13.     # write_csv()
  14.     # read_csv()
  15.     get_china_code()

image.png

3. 将列表转换为字符串

因为从csv文件中读出来的数据 默认是列表格式。因为前6位希望是字符串 格式。

join 方法列表转换 字符串

  2. code = ['659000']
  3. print(code,type(code)) # # 659000 <class 'str'>
  4. # 转换 字符串
  5. a = "".join(code)
  6. print(a,type(a)) # 659000 <class 'str'>

可以将列表转换 为字符串。
对应的代码

  1. def get_china_code():
  2.     #  as f 别名 f相当于是 前面open打开的文件  绝对路径字符串 前添加r
  3.     with open(file=r"C:\Users\zengy\PycharmProjects\pythonProject13\testdata\gb2260.csv",
  4.               encoding='utf8',mode='r') as f:
  5.         # 读取完成之后会将所有的内容放在一个列表对象中
  6.         rows = csv.reader(f)
  7.         # 循环列表
  8.         for row in rows:
  9.             # print(row,type(row))
  10.             code = "".join(row)
  11.             print(code,type(code))
  12.         # TODO 随机从这么多的地区码中选择一个 返回。
  13.         random_code = "659000"
  14.         return random_code
  15. if __name__ == '__main__':
  16.     # write_csv()
  17.     # read_csv()
  18.     get_china_code()

使用random.choice 获取随机

  1. import csv
  2. import random
  3. def get_china_code():
  4.     #  as f 别名 f相当于是 前面open打开的文件  绝对路径字符串 前添加r
  5.     with open(file=r"C:\Users\zengy\PycharmProjects\pythonProject13\testdata\gb2260.csv",
  6.               encoding='utf8',mode='r') as f:
  7.         # 读取完成之后会将所有的内容放在一个列表对象中
  8.         rows = csv.reader(f)
  9.         # 定义空列表
  10.         allcodes = []
  11.         # 循环列表
  12.         for row in rows:
  13.             # print(row,type(row))
  14.             code = "".join(row)
  15.             # 每次取到一个数据，讲这个数据放在列表中
  16.             allcodes.append(code)
  17.             # print(code,type(code))
  18.         # for循环执行完，查看列表的内容
  19.         print(allcodes)
  20.         # 随机从这么多的地区码中选择一个 返回。
  21.         random_code = random.choice(allcodes)
  22.         return random_code
  23. if __name__ == '__main__':
  24.     # write_csv()
  25.     # read_csv()
  26.     code = get_china_code()
  27.     print(f"获取随机code值：{code}")

代码整理

根据单一性原则，将获取code的代码放在 utils.py 工具模块中。
整理之后的代码，

  1. """
  2. 这个文件中存放自己定义的一些常用的工具函数。
  3. 生成测试数据
  4. """
  5. import random
  6. import csv
  8. def get_china_code():
  9.     #  as f 别名 f相当于是 前面open打开的文件  绝对路径字符串 前添加r
  10.     with open(file=r"C:\Users\zengy\PycharmProjects\pythonProject13\testdata\gb2260.csv",
  11.               encoding='utf8',mode='r') as f:
  12.         # 读取完成之后会将所有的内容放在一个列表对象中
  13.         rows = csv.reader(f)
  14.         # 定义空列表
  15.         allcodes = []
  16.         # 循环列表
  17.         for row in rows:
  18.             # print(row,type(row))
  19.             code = "".join(row)
  20.             # 每次取到一个数据，讲这个数据放在列表中
  21.             allcodes.append(code)
  22.             # print(code,type(code))
  23.         # for循环执行完，查看列表的内容
  24.         # print(allcodes)
  25.         # 随机从这么多的地区码中选择一个 返回。
  26.         random_code = random.choice(allcodes)
  27.         return random_code
  29. def get_phone():
  30.     """
  31.     自动随机生成一个手机号码
  33.     :return:
  34.     """
  35.     pre_phones = ["130", "131", "132", "133", "134", "135", "136", "137", "138", "139",
  36.                   "170", "172", "179",
  37.                   "150", "151", "155", "156",
  38.                   "188",
  39.                   "199"]
  40.     # 随机取号段
  41.     pre_3 = random.choice(pre_phones)
  42.     # print(f"随机选择号段 {pre_3}")
  43.     # 生成后8位
  44.     last_8 = ""
  45.     for i in range(8):
  46.         # 随机生成一个数字n
  47.         n = random.randint(0, 9)
  48.         # 拼接字符串 将n转换为字符串
  49.         last_8 = last_8 + str(n)  # 每次随机取到一个数字，转换字符串之后 放到last_8 后面
  50.     # print(f"生成后8位 {last_8}")
  51.     # 将前3位后8位 组合一起
  52.     phone = pre_3 + last_8
  53.     return phone
  55. def get_id():
  56.     """
  57.     随机生成身份证号
  59.     :return:
  60.     """
  61.     # 函数定义在函数的内部  闭包
  62.     def pre_6():
  63.         n = get_china_code()
  64.         # n = ""
  65.         # for i in range(6):
  66.         #     # 循环6次 每次生成1个随机数字，转换位字符串进行拼接
  67.         #     n = n + str(random.randint(0, 9))
  68.         # 循环执行完成
  69.         return n
  71.     def ymd_8(startyear=1922, endyear=2021):
  72.         # 生成出生的年  随机生成
  73.         year = random.randint(startyear, endyear)
  74.         # print(f"生成年份 {year}")
  75.         # 生成月份  1-12
  76.         month = random.randint(1, 12)
  78.         # 如果月份在1-9 之间，前面需要补0
  79.         if month < 10:
  80.             month = "0" + str(month)  # 转换位字符串之后前面 补0
  81.         # print(f"生成的月份 {month}")
  82.         # 1，3，5，7，8，10，12 天数 31天
  83.         if int(month) in [1, 3, 5, 7, 8, 10, 12]:
  84.             day = random.randint(1, 31)  #
  85.         # 4,6,9,11 月 30天
  86.         elif int(month) in [4, 6, 9, 11]:
  87.             day = random.randint(1, 30)
  88.         # 闰年 2月份 29天
  89.         elif (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
  90.             day = random.randint(1, 29)
  91.         # 以上都不符合，那就是平年
  92.         else:
  93.             day = random.randint(1, 28)
  95.         # 如果day 在1-9 之间 前面补0
  96.         if day < 10:
  97.             day = "0" + str(day)
  98.         # print(f"生成的日期 {day}")
  99.         # 转换为字符串 进行拼接
  100.         ydm = str(year) + str(month) + str(day)  # 等价于 f"{year}{month}{day}"
  101.         return ydm
  103.     def last_4():
  104.         nums = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'X']
  105.         # 前三位 数字
  106.         n3 = ""
  107.         for i in range(3):
  108.             n3 = n3 + str(random.randint(0, 9))
  109.         # 第四位 随机从 nums 选择一个
  110.         n4 = random.choice(nums)
  111.         # 将n3 和 n4 拼接之后返回
  112.         return n3 + n4
  114.     # 生成身份证id
  115.     id = pre_6() + ymd_8() + last_4()
  117.     return id
  121. if __name__ == '__main__':
  122.     # 测试生成手机号
  123.     # print(get_phone())
  124.     # 测试生成身份证id
  125.     print(get_id())

file_handler中只保存 读取和写入相关的操作。

  1. """
  2. 处理文件
  3. """
  4. # 导入csv
  5. import csv
  6. from common.utils import get_id,get_phone
  8. def write_csv():
  9.     with open('data.csv',encoding='utf8',mode='w',newline="") as f:
  10.         # 使用 csv的方式来写入， 固定用法
  11.         wr = csv.writer(f)
  12.         # 开始写入数据 往文件中写入数据， 数据必须是 列表格式。写入一行
  13.         wr.writerow(["手机号","身份证id"])
  14.         for i in range(10):
  15.             wr.writerow([get_phone(),get_id()])
  17. def read_csv():
  18.     #  as f 别名 f相当于是 前面open打开的文件
  19.     with open(file="data.csv",encoding='utf8',mode='r') as f:
  20.         # reader 读取文件内容 固定用法
  21.         all_rows = csv.reader(f)
  22.         # csv 中有很多行数据，需要一行一行去读
  23.         for line in all_rows:
  24.             print(line)
  27. if __name__ == '__main__':
  28.     write_csv()
  29.     # write_csv()
  30.     # read_csv()
  31.     # code = get_china_code()
  32.     # print(f"获取随机code值：{code}")

所有代码

2022-05-19-project.zip