将项目添加到列表列表 - Python

Question

我希望使用此处的代码将域与其 DNS 解析器名称相匹配

当前 CSV 输出

domain1 dns1 dns2 dns3 \n domain2 dns1 dns2 dns3 \n etc

这是不正确的格式，因为它将所有域和 DNS 解析器添加到同一行，而不是基于新域的新行。由于换行符，它们仅由空白单元格分隔。相反，我希望将其编写如下，其中每个域及其 dns 解析器都写入其自己的单独行。

预期的 CSV 输出：

domain1 dns1 dns2 dns3

domain2 dns1 dns2 dns3

domain3 dns1 dns2 dns3

etc...

我希望 CSV 文件以正确的格式写出，并且使用我拥有的代码，每次将域传递给 def dns_resolver 时，它都应该迭代到新的列表索引。这样，每个域及其 dns 解析器都有自己的列表，因此在写入新的 CSV 文件时，每个域将打印在 CSV 文件中自己的行中。

该代码未正确迭代列表索引，因此不会将域及其 dns 名称添加到任何列表中。当它们全部写入同一个列表时，它工作正常，但它们全部写入同一行，这是不正确的。因此，我不会使用 1 个列表，而是使用一个列表列表，并将每个列表写入自己的列表，然后将每个列表写入 csv 文件，以便它们位于自己的行中。通常，域将从 csv 文件读入列表，但为此，我输入了 3 个值。

import dns.resolver 

import csv

import os

from os.path import dirname, abspath

r = 0

def dns_resolver(domain):

    server = []

    resolvers = []

    

    try:

        resolvers = dns.resolver.resolve(domain, 'NS')

        #dns_list.append(domain)

        for x in resolvers:

            #dns_list.append(x.target)

            #dns_list.append('\n') 

            server.append(str(x.target))

    except:

        server.append('did not resolve')

    finally:

        return (domain, *server)

            

# Read in all domains from csv file domains.csv & count how many domains there are listed

domain_list = ['google.com', 'facebook.com', 'github.com']

domain_amount = 0

with open(domainFName, 'r') as file:

    for line in csv.reader(file):

        name = (line)

        domain_list.append(line)

        domain_amount += 1

for first_domain in domain_list:

    for x in first_domain:

        outputWriter.writerow(dns_resolver(x))

Answer 1

您可以简单地让dns_resolver函数返回给定域的列表。这*server是将每个项目附加到列表中的简写。使用列表理解，将所有列表收集到列表列表中以写入 CSV。def dns_resolver(domain):    # do your dns resolution    # server = dns.resolver.resolve(domain, 'NS')    server = ["dns1", "dns2", "dns3", "dns4"]    return [domain, *server]        # Read in all domainsdomain_list = ['google.com', 'facebook.com', 'github.com']print([dns_resolver(d) for d in domain_list])输出：[   ['google.com', 'dns1', 'dns2', 'dns3', 'dns4'],   ['facebook.com', 'dns1', 'dns2', 'dns3', 'dns4'],   ['github.com', 'dns1', 'dns2', 'dns3', 'dns4']]