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

Question

I have a list with many items in each entry delimited by [] . eg,

['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']

I want to find entries that contain this regex, and then capture the entire row in a variable:

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

I have unsuccesfully tried:

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)

I get a _csv.Error: iterator should return strings, not list (did you open the file in text mode?) .

Argh.

Answer 1

First, import_csv should not wrap the list in another list, it should just return the list of rows.

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

Second, exclude_entries doesn't need to use csv , that was already used when the data was imported, and name_entries is the list of rows.

Third, you should only be matching against the list element that contains the email address.

You can use filter() to filter the list, rather than a loop.

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)