切片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'] 

的切片表示法的全面概述見Explain Python's slice notation。