python，返回的位置和文件大小

Question

函数返回主井在python中的位置。

def myPath():
    for root,dirs,files in os.walk(dir):
        for fn in files:
            path = os.path.join(root, fn)
            return path
        return path
    return path

ls /home/bb/C/
a.out  main.c  simple_write  t.c

三个位置"；返回"；不是我想要的

我想得到结果"C"中的所有文件；

def filesize(path):
    for root, dirs, files in os.walk(PATH):
        for fn in files:
            path = os.path.join(root, fn)
            size = os.stat(path).st_size
            yield size,path
for size,path in filesize(PATH):
        print size,path

但是，如何用以下代码实现上述功能？如何修改？

def find(path):
    return [os.path.join(root,fn)
        for root,dir,files in os.walk(dirs)
        for fn in files]

Answer 1

返回路径的列表，而不仅仅是单个路径：

def find(path):
    return [os.path.join(root, fn)
            for root, dirs, files in os.walk(path)
            for fn in files]

您也可以在内部循环中使用yield来获得生成器函数，请参阅解释的Python yield关键字。

Answer 2

您想要的是一个生成器：

def myPath(mydir):
    for root,dirs,files in os.walk(mydir):
        for fn in files:
            path = os.path.join(root, fn)
            yield path # <----- Instead of return

for path in myPath(some_dir):
    print path

Answer 3

如果您只需要/home/bb/C目录列出的，这里有三种方法

import os, glob
def listWithoutDirectories(path):
    return [os.path.join(root,fn)
        for root,dirs,files in os.walk(mydir)
            if(root==path)
                for fn in files]
def listWithDirectories(path)
    return [os.path.join(path,fn)
        for fn in os.listdir(path)]

print listWithoutDirectories('/home/bb/C')
print listWithDirectories('/home/bb/C')
#Or with a glob
print glob.glob('/home/bb/C/*')