切片python os.listdir()

时间:2011-06-07 19:40:30

标签: python slice

我可以在os.listdir()中进行切片吗?只采取一些要素。

2 个答案:

答案 0 :(得分:3)

我不明白为什么不:

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

答案 1 :(得分:3)

由于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