搜索包含部分名称的文件并添加到字典

Question

大家好，我有文件保存在某个目录中。

dir ="/user/raw/samples"

我需要将所有以特定字符串结尾的文件放入字典 dic_x 并将其他文件放入字典 dic_z.

以下是 dir 中的文件名：

SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L002_R1_001_MM_1.fz.gz
SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L002_R2_001_MM_1.fz.gz
SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L001_R2_001_MM_1.fz.gz
SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L001_R1_001_MM_1.fz.gz

我有两本词典

 import sys
import os
import subprocess
import glob
import fnmatch
import re
    listF = ['ALPH']
    FH_sample = listF
    dic_x = {}
    dic_z = {}

    for line in FH_sample:
        dic_x[line.strip().split('\n')[0]] =[]
        dic_z[line.strip().split('\n')[0]] =[]

        for root, sub, file in os.walk(dir):
            for name in file:
                if name.endswith(".fz.gz"):
                    m1 = re.search('(.*)_R1', file)
                    m2 = re.search('(.*)_R2', file)
                if m1 and m1.group(1) in dic_x:
                    dic_x[m1.group(1)].append(file)
                if m2 and m2.group(1) in dic_z:
                    dic_z[m2.group(1)].append(file)

但是，我无法使用上面的代码行获取字典中的所有值。我需要像下面这样的词典，

dic_x = {'ALPH': ['SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L001_R1_001_MM_1.fz.gz','SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L002_R1_001_MM_1.fz.gz']}

dic_z = {'ALPH': ['SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L001_R2_001_MM_1.fz.gz','SAP_XCV_1980_H7LYVDRXX_2_ALPH_C_cR_c1_v1_0_FCDSSA_DDDD_S14_L002_R2_001_MM_1.fz.gz']}

目前 dic_x 和 dic_z 看起来像这样，

dic_z = {'ALPH': []}
dic_x = {'ALPH': []}

Answer 1

你可以利用列表

import os
import fnmatch
path = "test_images/"
x_list = ([f for f in os.listdir(path) if fnmatch.fnmatch(f, "*.png")])
z_list= ([f for f in os.listdir(path) if fnmatch.fnmatch(f, "*.bmp")])
dic_x = {'ALPH' : x_list }
dic_z = {'ALPH': z_list}

在您的情况下，您可以将 *.png 替换为 *R1_001_MM_1.fz.gz，将 *.bmp 替换为 *R2_001_MM_1.fz.gz'。还有 test_images/ 和 /user/raw/samples/

搜索包含部分名称的文件并添加到字典

Search for file with part of its name and add to dictionary

python

dictionary

os.walk