数学建模社区-数学中国

标题: 使用python 找到指定后缀的文件列表 [打印本页]
作者: 2744557306    时间: 2024-3-31 17:13
标题: 使用python 找到指定后缀的文件列表
找到指定后缀的文件，返回找到的文件路径列表，会递归文件夹。
  1. import os
    8 T# Z1 M# w( S6 w, l

  2. + R2 M3 y; U' Z1 ~( U. x7 `

  3. % F' C" Y6 i5 E8 j) D! @' s

  4.   C1 \' n' v9 H9 E
  5. # 找到指定后缀的文件
    2 O/ T" U0 s/ T' z) L( F
  6. def find_type(path:str,fix:str):3 e# `/ {5 O' U/ v/ \& o# {6 [
  7.     dlist=os.listdir(path)
    1 W. {1 D& ?/ ~# r6 b7 ^
  8.     file_list=[]& M" c0 B, n& \3 n4 o* u
  9.     for i in dlist:2 h: I1 s1 o. N: G, L4 J
  10.         ps=os.path.join(path, i)
    + y& q% O( e; `/ Q
  11.         if os.path.isdir(ps):
    - P- n  F4 J: K) z  W. x. V  B) v
  12.             file_list+=find_type(ps,fix)4 u/ Q- c; P# o0 G& F+ E
  13.         else:' m* b8 B; e0 X) H3 R; t
  14.             if(ps[-len(fix):]==fix):
    1 D7 t3 ~+ Q+ `+ ~9 L4 Q# h+ `
  15.                 file_list.append(ps)
    8 W: U$ l, d5 D2 U+ q3 @$ t
  16.     return file_list
    # F3 j7 ?* k, G  Z' B1 T+ `
  17. $ h* }! \' p( r# U1 G' e4 T
复制代码

, `3 ?/ d9 q9 ]: {6 }" C: ]



欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5