comparer deux listes dans python et retourner des indices de valeurs correspondantes

Pour les deux listes a et b, comment puis-je obtenir les indices de valeurs qui apparaissent dans les deux? Par exemple,

a = [1, 2, 3, 4, 5]
b = [9, 7, 6, 5, 1, 0]

return_indices_of_a(a, b)

serait de retour [0,4]avec (a[0],a[4]) = (1,5).

source d'informationauteur user1342516

  1. 19

    La meilleure façon de le faire serait de faire b un set puisque vous êtes la seule vérification de l'appartenance à l'intérieur.

    >>> a = [1, 2, 3, 4, 5]
>>> b = set([9, 7, 6, 5, 1, 0])
>>> [i for i, item in enumerate(a) if item in b]
[0, 4]
  2. 5
    def return_indices_of_a(a, b):
  b_set = set(b)
  return [i for i, v in enumerate(a) if v in b_set]
  3. 2

    Pour les grandes listes cela peut être de l'aide:

    for item in a:
index.append(bisect.bisect(b,item))
    idx = np.unique(index).tolist()

    Assurez-vous d'import numpy.