dandan0523 发表于 2021-3-25 10:56:59

麻烦各位看个代码


题目：编写一个程序，统计当前目录下每个文件类型的文件数
import os

type_dict=dict()

all_files=os.listdir('C:\\clear')

for each_file in all_files:
    if os.path.isdir(each_file):
      type_dict.setdefault('文件夹',0)
      type_dict['文件夹']+=1
    else:
      (f_name,f_extension)=os.path.splitext(each_file)
      ext=f_extension
      type_dict.setdefault(ext,0)
      type_dict+=1



for each_type in type_dict.keys():
    print('该文件夹下共有类型为【%s】的文件%d个'%(each_type,type_dict))





运行结果：
该文件夹下共有类型为【.py】的文件1个
该文件夹下共有类型为【.txt】的文件1个
该文件夹下共有类型为【】的文件2个
该文件夹下共有类型为【.rmvb】的文件1个
该文件夹下共有类型为【.mp4】的文件1个
该文件夹下共有类型为【.avi】的文件1个


问题：这里不应该显示【文件夹】吗？

hrp 发表于 2021-3-25 11:15:25

本帖最后由 hrp 于 2021-3-25 11:20 编辑

不是'文件夹'的原因是each_file是个文件夹，但程序执行了else分支，也就是isdir(each_file)返回了False，为什么会出现这个情况？因为each_file只是个文件夹或者文件名，不是完整路径，所以isdir(each_file)判断的时候，isdir会把each_file当成是当前文件夹(你的py文件所在目录)下的文件，而不是'C:\\clear'文件夹下的文件。所以要正常工作，要么用os.chdir把当前文件夹设置为'C:\\clear'，要么给os.isdir传完整路径。

#未测试
import os

type_dict=dict()

mpath = 'C:\\clear'

os.chdir(mpath)
all_files=os.listdir(mpath)

for each_file in all_files:
    if os.path.isdir(each_file):
      type_dict.setdefault('文件夹',0)
      type_dict['文件夹']+=1
    else:
      (f_name,f_extension)=os.path.splitext(each_file)
      ext=f_extension
      type_dict.setdefault(ext,0)
      type_dict+=1



for each_type in type_dict.keys():
    print('该文件夹下共有类型为【%s】的文件%d个'%(each_type,type_dict))

dandan0523 发表于 2021-3-25 15:46:46

谢谢您，回答得太详细了{:5_91:}
页: [1]
查看完整版本: 麻烦各位看个代码