python 中的 os.walk 未使用作为路径传递的命令行参数运行

Question

我需要查找系统上文件夹中的文件数量。

这就是我使用的：

file_count = sum((len(f) for _, _, f in os.walk('path')))

当我们将路径指定为引号中的字符串时，效果很好，但是当我输入保存路径的变量名时， type(file_count) 是一个生成器对象，因此不能用作整数。

如何解决这个问题以及为什么会发生这种情况？

好的，这就是我正在做的：

在终端的命令行中：

python mypyProg.py arg1 arg2 arg3

在 myProg.py 中：

arg1 = sys.argv[1]
file_count = sum((len(f) for _, _, f in os.walk(arg1)))

arg1 作为字符串传递

我检查了 repr(arg1) 和 type(arg1):

repr(arg1) '/home/kartik/Downloads/yahoo_dataset/tryfolder'
type(arg1) <type 'str'>

type(file_count) <type 'generator'>

错误消息：

NDCG = scipy.zeros((file_count,1),float)

    TypeError: an integer is required

我不知道，它当我仅使用一些虚拟变量进入 IDLE python IDE 时，它运行良好。

Answer 1

我假设您使用 walk 是因为您想了解目录及其子目录中的每个文件。我不明白这里发生了什么：

file_count = sum((len(f) for _, _, f in os.walk(path)))

假设路径包含“src”，它是我的主目录中的一个目录我得到了目录及其后代中的文件数量，那么你是什么意思？您确定从命令行正确读取了路径吗？可以多发帖吗？

Answer 2

sum 应该返回一个整数，就像在我的 python shell 中一样......

>>> x = '/tmp'
>>> file_count = sum((len(f) for _, _, f in os.walk(x)))
>>> file_count
11
>>> type(file_count)
<type 'int'>

Answer 3

由于我在该目录中只有文件，因此我使用了这个：

file_count = len([f for f in os.listdir(loadpathTest) if os.path.isfile(os.path.join(loadpathTest, f))])

这似乎有效。

@全部
感谢您的帮助。