Question

如何在主文件夹和子文件夹中打印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] )

第一个问题解决了一个错误

第二个问题：代码输出失败

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

我希望得到输出

 aux
 tex
 pdf
 php
 ...

Answer 1

让我稍微解释调试方法。

当你遇到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()

Answer 2

if len(sys.argv) != 1:

我认为你的意思是2。 argv[0]是脚本的名称; argv[1]是第一个参数，等等。

Answer 3

正如其他人所说，sys.argv的第一个元素是程序::

# argv.py
import sys

for index, arg in enumerate(sys.argv):
    print '%(index)s: %(arg)s' % locals()

如果我在没有参数::

的情况下运行它

$ python argv.py 
0: argv.py

我看到argv中的第一个也是唯一一个项目是程序/脚本的名称。如果我传递参数::

$ python argv.py a b c
0: argv.py
1: a
2: b
3: c

等等。

另一件事是你真的不需要在os.walk ::

产生的元组的第三个元素中的项目上使用os.path.basename

import os
import sys

# Imagine some usage check here...

# Slice sys.argv to skip the first element...
for path in sys.argv[1:]:
    for root, dirs, files in os.walk(path):
        for name in files:
            # No need to use basename, since these are already base'd, so to speak...
            print name

Answer 4

sys.argv的长度为2，因为索引为0的项目（程序名称）和索引为1的项目（程序的第一个参数）。

将程序更改为与2进行比较似乎可以给出正确的结果，而不进行任何其他更改。

Answer 5

argv通常包含作为第一个元素调用的程序/脚本的名称，因此传递单个参数时的长度实际上是2，而不是1。

通过Python打印基本名称

5 个答案: