高渐飞 发表于 2018-7-16 15:15:36

A-14-OS模块练习

#coding: UTF-8

#知识点总结
#1.os模块即操作系统模块,详细文档http://bbs.fishc.org/thread-45512-1-1.html
#2.


#实操练习
#1.查询文件夹下各种文件类型的个数
#思路-利用os.listdir得到指定文件夹下各文件的名称,
#之后通过循环,查找特定文件扩展名是否在已得到的文件名中,有则加一。最后输出即可。
#优点:代码简单,思路清晰;缺点:需提前指定文件扩展名,未指定的无法监测其个数
import os

file_name = os.listdir('E:\\test')
print(file_name)

txt = 0
png = 0
py = 0
docx = 0
doc = 0

for each_file in file_name:
    if '.txt' in each_file:
      txt += 1
    elif '.png' in each_file:
      png += 1
    elif '.py' in each_file:
      py += 1
    elif '.docx' in each_file:
      docx += 1
    elif '.' not in each_file:
      doc += 1

print('txt 是 %d, png 是 %d, py 是 %d, docx 是 %d, doc 是 %d' % (txt,png,py,docx,doc))


#2.改进版-查询文件夹下各种文件类型的个数
#思路-通过os.listdir得到文件名称,再对文件名称进行遍历,分离出名称和扩展名
#若扩展名在file_type字典中,其value值赋1;否则原value+1;之后进行下一次循环直到遍历结束
#最后根据键值即文件扩展扩展名,遍历字典file_type输出value即可
importos

defseek_extension(path):
    file_type = {}         #建立空字典,有key和value两部分
    file_names = os.listdir(path)            #列举指定目录中的文件名,如a.txt,b.py,c.txt
    foreach_file in file_names:
      file_name_extension = os.path.splitext(each_file)#分离文件名与扩展名,返回(f_name, f_extension)元组
      iffile_name_extension not in file_type:   #file_name_extension此处即f_extension。
                                                         #file_name_extension非元组嵌套元组,每次都是(name,ext...)
            file_type] = 1    #例如file_name_extension为txt,字典通过键值key(可为字符串)来精确索引
      else:
            file_type] += 1

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

path = input("输入要查找的绝对路径: ")
seek_extension(path)


#3.查询指定文件夹下各文件的大小
#思路-通过os.listdir获得指定目录下文件名,遍历文件,其中通过os.path.getsize即可获得文件大小。
#通过字典来存储文件名和其占用内存大小,遍历输出即可
import os

deffile_size_calc(path):
    file = os.listdir(path)   #得到文件名构成列表如['a.txt','b.py']
    file_size = {}            #key为文件名,value为文件大小,如{'asd.txt':141,'b.py':409}
    foreach_file in file:
      file_size = os.path.getsize(path + '\\' + each_file)   #其中path为绝对路径,返回文件大小,单位字节

    foreach_key in file_size.keys():
      print('%s 【%d bytes】' % ( each_key, file_size ) )


path = input("请输入要查找位置的绝对路径: ")
while not isabs(path)
{
    path = input("请输入要查找位置的绝对路径: ")
}
file_size_calc(path)


#4.将指定目录下指定格式的文件保存的文件中
import os

doc_path = input('请输入待查找的初始目录:')
while not os.path.isdir(doc_path):            #判断指定路径是否存在且是一个目录
    doc_path = input('输入错误,请重新输入待查找的初始目录:')

save_path=input('请输入文件保存目录:')
while not os.path.isdir(save_path):            #判断指定路径是否存在且是一个目录
    save_path = input('输入错误,请重新输入文件保存目录:')

doc_walk = os.walk(doc_path)   #遍历doc_path路径下所有的子目录,返回一个三元组:(路径, [包含目录], [包含文件])

file_path = []   #建立列表,存储文件地址

for each_part in doc_walk:
    for each_file in each_part:
      if '.avi' or '.mp4' or '.rmvb' in each_file.lower():
            file_path.append(os.path.join(each_part,each_file) + '\n' * 2)#将each_part,each_file组合成一个路径名
                                                                               #将路径添加到列在file_path中
print(file_path)

f = open(save_path+'\\content2.txt','w',encoding = 'GBK')
f.writelines(file_path)
f.close()


#5.没看懂,以后再说

import os

defpos_in_line(line, key):
    pos = []
    begin = line.find(key)

    whilebegin != -1:
      pos.append(begin + 1)
      begin = line.find(key, begin + 1)

    return pos

defpos_in_file(file, key):
    dict_key = dict()
    count = 0
    target_file = open(file)

    foreach_line in target_file:
      count += 1
      ifkey in each_line:
            pos = pos_in_line(each_line, key)
            dict_key = pos

    target_file.close()
    return dict_key

defprint_key(dict_key):
    keys = dict_key.keys()
    keys = sorted(keys)
    foreach_key in keys:
      print("关键字出现在第 %s 行, 第 %s 个位置" % ( each_key, dict_key ) )


defsearch_key(path, key):
    os.chdir(path)

    foreach_file in os.listdir(os.curdir):
      ifos.path.isdir(each_file):
            search_key(each_file, key)
            os.chdir(os.pardir)
      ifos.path.isfile(each_file):
            ifos.path.splitext(each_file) == '.txt':
                dict_key = pos_in_file(each_file, key)
                ifdict_key != {}:
                  print('==================================================================================')
                  print("在文件【%s】中找到关键字【%s】" %( (os.getcwd() + os.sep + each_file), key ) )
                  print_key(dict_key)


path = input("请输入要查找的路径: ")
key = input("请输入要查找的关键字: ")
search_key(path, key)

软糖别太软 发表于 2022-8-9 16:23:06

请教一下:

#2.改进版-查询文件夹下各种文件类型的个数

这个地方的path是统一的绝对路径吗?
页: [1]
查看完整版本: A-14-OS模块练习