python之通过 Python 打印碱基名称

如何通过 Python 在主文件夹和子文件夹中打印文件的基本名称？

我的尝试

#!/usr/bin/python     
 
import os     
import sys     
 
def dir_basename (dir_name):     
        for dirpath, dirnames, filenames in os.walk(dir_name): 
                for fname in filenames: 
                        print os.path.basename(fname)            // Problem here! 
 
if len(sys.argv) != 1: 
        u = "Usage: dir_basename <dir_name>\n" 
        sys.stderr.write(u)                        
        sys.exit(1)             
 
dir_basename ( sys.argv[1] ) 

第一个问题用 off-by-one-error 解决了

第二个问题:代码给我的输出不成功

man.aux 
about_8php.tex 
refman.pdf 
successful_notice.php 
... 

我希望得到输出

 aux 
 tex 
 pdf 
 php 
 ... 

请您参考如下方法:

让我稍微解释一下调试方法。

当您遇到 len(sys.argv) != 1 的情况时，您应该问问自己:“len(sys.argv) 的实际值是多少？ ？为什么会这样？”。答案是:

>>> len(sys.argv) 
2 
>>> sys.argv 
['/tmp/basename.py', '/path/to/home/Desktop/pgCodes/'] 

我想现在问题应该更清楚了。

编辑:为了解决您的第二个问题，您感兴趣的东西称为文件扩展名 或后缀，而不是基本名称。这是完整的解决方案:

import sys, os 
 
def iflatten(xss): 
    'Iterable(Iterable(a)) -> Iterable(a)' 
    return (x for xs in xss for x in xs) 
 
def allfiles(dir): 
    'str -> Iterable(str)' 
    return iflatten(files for path, dirs, files in os.walk(dir)) 
 
def ext(path): 
    'str -> str' 
    (root, ext) = os.path.splitext(path) 
    return ext[1:] 
 
def main(): 
    assert len(sys.argv) == 2, 'usage: progname DIR' 
    dir = sys.argv[1] 
 
    exts = (ext(f) for f in allfiles(dir)) 
    for e in exts: 
        print e 
 
if __name__ == '__main__': 
    main()