Python 以数字方式列出文件

Question

我有一组文件。

在每个文件中，有一列对应于一个文件编号。 它们都以这种格式命名：

test_file_page_#_out_of_7000.txt

例子：

test_file_page_1_out_of_7000.txt
test_file_page_2_out_of_7000.txt
test_file_page_7000_out_of_7000.txt

当我运行以下代码时： output_pages = os.listdir()

这些文件按以下顺序列出：

'test_file_Page_10_out_of_7000.txt', 'test_file_Page_11_out_of_7000.txt', ..., 'test_file_Page_1_out_of_7000.txt', 'test_file_Page_2_out_of_7000.txt'

我想要按此顺序列出的文件：

'test_file_Page_1_out_of_7000.txt', 'test_file_Page_2_out_of_7000.txt', ..., 'test_file_Page_10_out_of_7000.txt', 'test_file_Page_11_out_of_7000.txt'

有什么建议么？

Answer 1

sorted方法接受一个key参数，它是一个 function，它接受一个列表元素并返回应该用于排序的值：

def get_number(filename):
    return int(filename.split("_")[3])
    
files = sorted(os.listdir(), key=get_number)