读取目录中的文件子集

这应该可以为您完成以下工作:

import argparse
import re
from os import listdir

if __name__ == "__main__":

    parser = argparse.ArgumentParser(description='Process some files.')
    parser.add_argument("--range", type=str, help='file range (i.e. 0-100)', default='0-100')
    args = parser.parse_args()

    try:
        file_range = list(map(int, args.range.split('-')))
    except Exception as e:
        raise Exception('Invalid parameter format...\n%s' % e)

    dir_files = listdir('.')
    matches_list = list(map(lambda x: re.match(r'm([0-9]+)\.data', x), dir_files))
    files = [x.group() for x in matches_list if x and (file_range[0] <= int(x.groups()[0]) <= file_range[1])]
    print(files)

命令应该如下所示:

python script.py--范围0-200

一种解决方案是在命令行中传递两个数字来告诉范围:

选择Py

#!/usr/bin/env python3
import re
import sys
from pathlib import Path

def main():
    first, last = sys.argv[1:]
    pattern = re.compile(rf'tx0?[{first}-{last}]\.data$')
    # pattern = re.compile(r'tx0?[{}-{}]\.data$'.format(first, last))
    for p in Path().glob('*'):
        if pattern.match(p.name):
            print(p.name)


if __name__ == '__main__':
    main()

$python3 choose.py 2 4年

tx02.data
tx03.data
tx04.data