这个我应该怎么写呢用python

查找给定的目录 形如“/Users/dliu” 下是否包含有指定类型（形如：.zip）的文件

file_dir = “/Users/dliu”
for file in os.listdir(file_dir):
    if file.endswith('.zip'):
        print(file)