Python实现在当前目录新建目录img, 里面包含100个文件, 100个文件名 各不相同(X4G5.png)

题目要求：
1. 在当前目录新建目录img, 里面包含100个文件, 100个文件名
各不相同(X4G5.png)
2. 将当前img目录所有以.png结尾的后缀名改为.jpg.

代码块：

import random
import string
import os
from os.path import splitext

def gen_code(len=4):
    #随机生成4位验证码
    li = random.sample(string.ascii_letters+string.digits,len)
    #拼接为字符串
    return ''.join(li)

def create_files():
    #随机生成100个验证码
    li = [gen_code() for i in range(100)]
    os.mkdir('img')
    for i in li:
        os.mknod('img/' + i + '.png')

create_files()

def modify_suffix(dirname,old_suffix,new_suffix):
    #找出以png结尾的文件名
    # pngfile = [filename for filename in os.listdir(dirname) if filename.endswith(old_suffix)]
    pngfile = filter(lambda filename:filename.endswith(old_suffix),os.listdir(dirname))

    #分离文件名和后缀
    basefiles = [os.path.splitex