searching within nested list in python

Question

I have a list:

l = [['en', 60, 'command'],['sq', 34, 'komand']]

I want to search for 'komand' or 'sq' and get l[1] returned.

Can I somehow define my own matching function for list searches?

Answer 1

You can do it this way:

def find(value, seq):
    for index, item in enumerate(seq):
        if value in item: 
            return index, item

In [10]: find('sq', [['en', 60, 'command'],['sq', 34, 'komand']])
Out[10]: (1, ['sq', 34, 'komand'])

Or if you want a general solution:

def find(fun, seq):
    for index, item in enumerate(seq):
        if fun(item): 
            return index, item

def contain(value):
    return lambda l: value in l

In [14]: find(contain('komand'), [['en', 60, 'command'],['sq', 34, 'komand']])
Out[14]: (1, ['sq', 34, 'komand'])

Answer 2

If all you're trying to do is return the first list that contains a match for any of the values then this will work.

def search(inlist, matches):
    for li in inlist:
        for m in matches:
            if m in li:
                return li
    return None

>>> l = [['en', 60, 'command'],['sq', 34, 'komand']]
>>> search(l, ('sq', 'komand'))
['sq', 34, 'komand']

Answer 3

Yes:

has_oneof = lambda *patterns: lambda: values any(p in values for p in patterns)
result = itertools.ifilter(has_oneof('komand', 'sq'), l).next()
print result # prints ['sq', 34, 'komand']

Answer 4

An expression like:

next(subl for subl in l if 'sq' in subl)

will give you exactly the sublist you're searching for (or raise StopIteration if there is no such sublist; if the latter behavior is not what you want, pass next a second argument [[e.g, [] or None, depending on what exactly you want!]] to return in that case). So, just use this result value, or assign it to whatever name you wish, and so forth.

Of course, you can easily dress this expression up into any kind of function you like, e.g.:

def gimmethesublist(thelist, anitem, adef=None):
    return next((subl for subl in thelist if anitem in subl), adef)

but if you're working with specific variables or values, coding the expression in-line may often be preferable.

Edit: if you want to search for multiple items in order to find a sublist containing any one (or more) of your items,

its = set(['blah', 'bluh'])
next(subl for subl in l if its.intersection(subl))

and if you want to find a sublist containing all of your items,

next(subl for subl in l if its.issubset(subl))