python os.listdir() 的切片

Question

我可以在 os.listdir() 中進行切片嗎？ 只取一些元素。

Answer 1

我不明白為什么不：

>>> os.listdir(os.getcwd())
['CVS', 'library.bin', 'man', 'PyLpr-0.2a.zip', 'pylpr.exe', 'python26.dll', 'text']
>>> os.listdir(os.getcwd())[3:]
['PyLpr-0.2a.zip', 'pylpr.exe', 'python26.dll', 'text']

Answer 2

由於os.listdir(path)返回一個列表，因此您可以對結果使用切片表示法。 例如，您可以使用os.listdir(path)[:5]獲取前 5 個結果：

>>> import os
>>> os.listdir('.')
['f1', 'f10', 'f2', 'f3', 'f4', 'f5', 'f6', 'f7', 'f8', 'f9']
>>> os.listdir('.')[:5]
['f1', 'f10', 'f2', 'f3', 'f4']

有關切片表示法的全面概述，請參閱解釋 Python 的切片表示法。