Obtenir le premier index de liste contenant une sous-chaîne?

Question

Pour les listes, la méthode list.index(x) renvoie l'index dans la liste du premier élément dont la valeur est x. Mais si je veux regarder à l'intérieur des éléments de la liste, et pas seulement à l'ensemble des éléments, comment puis-je faire la plus grande méthode Pythoninc pour cela?

Par exemple, avec

l = ['the cat ate the mouse', 
    'the tiger ate the chicken', 
    'the horse ate the straw'] 

cette fonction retournerait 1 fournie avec l'argument tiger.

Answer 1

Méthode non slicky:

def index_containing_substring(the_list, substring): 
    for i, s in enumerate(the_list): 
     if substring in s: 
       return i 
    return -1 

Answer 2

def find(l, s): 
    for i in range(len(l)): 
     if l[i].find(s)!=-1: 
      return i 
    return None # Or -1 

Answer 3

Ceci est assez lisse et assez efficace.

>>> def find(lst, predicate): 
...  return (i for i, j in enumerate(lst) if predicate(j)).next() 
... 
>>> l = ['the cat ate the mouse','the tiger ate the chicken','the horse ate the straw'] 
>>> find(l, lambda x: 'tiger' in x) 
1 

Le seul problème est qu'il soulèvera StopIteration si l'élément est introuvable (bien que cela soit facile d'y remédier).

Answer 4

def first_substring(strings, substring): 
    return min(i for i, string in enumerate(strings) if substring in string) 

Note: Cela soulèvera ValueError si aucune correspondance est trouvée, ce qui est mieux à mon avis.

Answer 5

Variation de la solution abyx (optimisé pour arrêter quand le match se trouve)

def first_substring(strings, substring): 
    return next(i for i, string in enumerate(strings) if substring in string) 

Si vous êtes pré 2.6, vous aurez besoin de mettre le next() à la fin

def first_substring(strings, substring): 
    return (i for i, string in enumerate(strings) if substring in string).next() 

Answer 6

>>> li = ['my','array','with','words'] 
    >>> reduce(lambda tup, word: (tup[0], True) if not tup[1] and word == 'my' else (tup[0]+1 if not tup[1] else tup[0], tup[1]), li, (0, False))[0] 
    0 
    >>> reduce(lambda tup, word: (tup[0], True) if not tup[1] and word == 'words' else (tup[0]+1 if not tup[1] else tup[0], tup[1]), li, (0, False))[0] 
    3 

Answer 7

Vous pouvez utiliser le doublure suivante:

index = [idx for idx, s in enumerate(l) if 'tiger' in s][0]