python - 想在 python 中获取主目录中的文件数

Question

我看到了这个How to count the number of files in a directory using Python

并拥有这个:

import os, os.path

print len([name for name in os.listdir(os.path.expanduser("~")) if os.path.isfile(name)])

但它总是返回 0。我该如何修改它以返回文件数？

谢谢

Answer 1

此刻，您正在调用 os.path.isfile("somefile.ext")。您需要调用 os.path.isfile("~/somefile.ext")。

import os

homedir = os.path.expanduser("~")
print len([
    name
    for name in os.listdir(homedir)
    if os.path.isfile(os.path.join(homedir, name))
])

或者更简洁:

print sum(
    os.path.isfile(os.path.join(homedir, name)) for name in os.listdir(homedir)
)