选择一个文件夹，遍历文件夹内所有png格式图片，输出图片数量

import os
def list_image(path):
    ans = 0
    for file in os.listdir(path):
        if (os.path.splitext(file)[-1] == '.png'):
            print(file)
            ans += 1
    print(ans)

list_image('/Users/AvaterXXX/Desktop/AI_Lab/Assignment3/Images/')