比较列表和字典并将其打印到 YML 文件

Question

我有列表和字典，现在我需要将字典中的键与列表中的键进行比较，并将匹配结果打印到 YML 文件中。 所以这是我的字典键，

samples1.keys()
dict_keys(['C3N-02289_10_L1', 'C3N-02289_10_L2', 'C3N-02671_08_L1', 'C3N-02671_08_L2','C3N-02671_10_L1','C3N-02671_10_L1' ])

我有 2 个列表，

Left_reads = [
    'C3N-02289_10_L1_R1.gz',
    'C3N-02289_10_L2_R1.gz',
    'C3N-02671_08_L1_R1.gz',
    'C3N-02671_08_L2_R1.gz',
    'C3N-02671_10_L1_R1.gz',
    'C3N-02671_10_L2_R1.gz'
 ]



 Right_reads = [
    'C3N-02289_10_L1_R2.gz',
    'C3N-02289_10_L2_R2.gz',
    'C3N-02671_08_L1_R2.gz',
    'C3N-02671_08_L2_R2.gz',
    'C3N-02671_10_L1_R2.gz',
    'C3N-02671_10_L2_R2.gz'
]

现在我需要将字典 (samples1.key) 中的每个键与列表中的值 (Left_reads 和 Right_reads) 进行比较。如果密钥与列表中的字符串匹配，则将其打印到 YML 文件。 这是我尝试过的，

for sam in samples1.keys():
    ymlFile = pat + sam + '.yml'
    ymlFH = open(ymlFile, 'w')

    ymlFH.write("reads1: [\n")
    for sam in sorted(Left_reads):
        ymlFH.write(" {class: File, path: " + path + '/' + sam + "}, \n")
    ymlFH.write("]\n")

    ymlFH.write("reads2: [\n")    
    for sam in sorted(Right_reads):
        ymlFH.write(" {class: File, path: " + path + '/' + sam + "}, \n")
    ymlFH.write("]\n")
    ymlFH.close()

这会将列表Left_reads 和Right_reads 中的所有值打印到reads1 和reads2。

我的目标是有一个输出来打印列表中的每个值，例如，对于文件C3N-02289_10_L1.yml，输出应该是这样的，所以这里我的所有值都与C3N-02289_10_L1和C@987654333匹配@。我需要脚本来比较前缀，也就是C3N-02289_10在列表和字典之间然后写入YML文件

reads1: [
 {class: File, path: /usr/path/C3N-02289_10_L1_R1.gz}, 
 {class: File, path: /usr/path/C3N-02289_10_L2_R1.gz},

]
reads2: [
 {class: File, path: /usr/path/C3N-02289_10_L1_R2.gz}, 
 {class: File, path: /usr/path/C3N-02289_10_L2_R2.gz},

]

字典中的所有值、列表中的匹配字符串等都是相同的。 使用我上面的一段代码，C3N-02289_10_L1.yml 的输出看起来像，

 reads1: [
     {class: File, path: /usr/path/C3N-02289_10_L1_R1.gz}, 
     {class: File, path: /usr/path/C3N-02289_10_L2_R1.gz},
    {class: File, path: /usr/path/C3N-02671_08_L1_R1.gz}, 
    {class: File, path: /usr/path/C3N-02671_08_L2_R.gz},
    {class: File, path: /usr/path/C3N-02671_10_L1_R1.gz},
    {class: File, path: /usr/path/C3N-02671_10_L2_R1.gz} , 
    ]
    reads2: [
     {class: File, path: /usr/path/C3N-02289_10_L1_R2.gz}, 
     {class: File, path: /usr/path/C3N-02289_10_L2_R2.gz},
    {class: File, path: /usr/path/C3N-02671_08_L1_R2.gz}, 
    {class: File, path: /usr/path/C3N-02671_08_L2_R2.gz},
    {class: File, path: /usr/path/C3N-02671_10_L1_R2.gz},
    {class: File, path: /usr/path/C3N-02671_10_L2_R2.gz} 
    ]

Answer 1

首先让我们从你的目标开始。

在您的代码中，字典中的键与列表中的值之间没有比较。据我了解，您想检查当前字典键是否是列表中值的前缀，如果是，则将该文件名转储到 .yaml 文件。

所以你的代码应该是这样的：

for prefix in samples1.keys():
    for filename in some_list:
        if filename.startswith(prefix):
            # add the {class: File, path: some/path/filename } to the yaml file

您的代码的第二个输出不是有效的 yaml 文件。 我推荐使用PyYaml 包。

如果我们把它们放在一起，我们会得到：

import yaml

# definition of path variable is here somewhere...

# edited to take only the prefixes of the keys
desired_keys = ['_'.join(k.split('_')[:-1]) for k in samples1.keys()]

for prefix in desired_keys:
    yml_filename = prefix + '.yaml'
    reads1 = []
    for filename in Left_reads:
        if filename.startswith(prefix):
            reads1.append({'class': 'File', 'path': path + '/' + filename})

    reads2 = []
    for filename in Right_reads:
        if filename.startswith(prefix):
            reads2.append({'class': 'File', 'path': path + '/' + filename})

    data = {'reads1': reads1, 'reads2': reads2 }
    stream = open(yml_filename, 'w')
    yaml.dump(data, stream)
    stream.close()

顺便说一句，我建议使用os.path.join(path, filename) 方法而不是path + '/' + filename，以减少出错的可能性。

编辑

使用给定的Left_reads、Right_reads 和samples1.keys()，结果是三个.yml 文件：

C3N-02289_10.yml C3N-02671_08.yml C3N-02671_10.yml

第一个，即C3N-02289_10.yml 包含：

reads1:
- {class: File, path: /path/yamlTest/__main__/C3N-02289_10_L1_R1.gz}
- {class: File, path: /path/yamlTest/__main__/C3N-02289_10_L2_R1.gz}
reads2:
- {class: File, path: /path/yamlTest/__main__/C3N-02289_10_L1_R2.gz}
- {class: File, path: /path/yamlTest/__main__/C3N-02289_10_L2_R2.gz}