如果列表中的项目在 python 中有某个字符串，我该如何提取它？

Question

我有一个列表，每个条目中有许多项目，由[]分隔。 例如，

['1', 'pbkdf2_sha256$100000$sk3ONL23432fsdgUsHM62xa9XJHL+LkJHhK3cFGj8LYWGtOd8HC7Hs=',
'2018-09-25 19:32:41', '0', '', '', 'bob@trellis.law', 'Bob', 'Simon', 
'bob@trellis.law', '1', '0', '2016-12-30 17:43:41', 'Bob Simon', 'Bob', '0', '1', 
'', '[]', '', '0', '1', '0', '1', '', '', '1', '14', '191', '1', '0', '1', '0', '', 
'', '', '0']

我想找到包含此正则表达式的条目，然后在变量中捕获整行：

r = re.compile(r'\w+\+\d+@trellis\.law')

我没有成功地尝试过：

def import_csv(csv_file):
    name_entries = []
    with open(csv_file) as csvfile:
        reader = csv.reader(csvfile)
        name_entries.append(list(reader))
    return name_entries


def exclude_regex_users(name_entries):
    pulled_names = []
    r = re.compile(r'\w+\+\d+@trellis\.law')

    reader = csv.reader(name_entries)

    for read in reader:
        n = r.match(read)
        if n:
            pulled_names.append(n.group())

    print(pulled_names)

我得到一个_csv.Error: iterator should return strings, not list (did you open the file in text mode?) 。

啊。

Answer 1

首先， import_csv不应该将列表包装在另一个列表中，它应该只返回行列表。

def import_csv(csv_file):
    name_entries = []
    with open(csv_file) as csvfile:
        reader = csv.reader(csvfile)
        return list(reader)

其次， exclude_entries不需要使用csv ，在导入数据时已经使用， name_entries是行列表。

第三，您应该只匹配包含 email 地址的列表元素。

您可以使用filter()过滤列表，而不是循环。

def exclude_regex_users(name_entries):
    r = re.compile(r'\w+\+\d+@trellis\.law')
    pulled_names = filter(lambda row: r.match(row[6]) or r.match(row[9]), name_entries)

    print(pulled_names)