Find the index of an item in a list of lists

by

I’d do something like this:

[(i, colour.index(c))
 for i, colour in enumerate(colours)
 if c in colour]

This will return a list of tuples where the first index is the position in the first list and second index the position in the second list (note: c is the colour you’re looking for, that is, "#660000").

For the example in the question, the returned value is:

[(0, 0)]

If you just need to find the first position in which the colour is found in a lazy way you can use this:

next(((i, colour.index(c))
      for i, colour in enumerate(colours)
      if c in colour),
     None)

This will return the tuple for the first element found or None if no element is found (you can also remove the None argument above in it will raise a StopIteration exception if no element is found).

Edit: As @RikPoggi correctly points out, if the number of matches is high, this will introduce some overhead because colour is iterated twice to find c. I assumed this to be reasonable for a low number of matches and to have an answer into a single expression. However, to avoid this, you can also define a method using the same idea as follows:

def find(c):
    for i, colour in enumerate(colours):
        try:
            j = colour.index(c)
        except ValueError:
            continue
        yield i, j

matches = [match for match in find('#660000')]

Note that since find is a generator you can actually use it as in the example above with next to stop at the first match and skip looking further.

More Related Contents:

  1. Replace list object based on indexes of another list (Python)
  2. Finding the index of an item in a list
  3. Finding the index of an item in a list
  4. How do I get the last element of a list?
  5. How to remove an element from a list by index
  6. Access multiple elements of list knowing their index
  7. How to find all positions of the maximum value in a list?
  8. In Python, how do I index a list with another list?
  9. Explicitly select items from a list or tuple
  10. Python: return the index of the first element of a list which makes a passed function true
  11. Tuple unpacking order changes values assigned
  12. Using an index to get an item
  13. Sort list by frequency
  14. Insert element in Python list after every nth element
  15. Understand Python swapping: why is a, b = b, a not always equivalent to b, a = a, b?
  16. Finding the indices of matching elements in list in Python
  17. Implement list-like index access in Python
  18. TypeError: ‘float’ object is not subscriptable
  19. Indexing a list with an unique index
  20. Python Bool and int comparison and indexing on list with boolean values
  21. Index all *except* one item in python
  22. Python IndexError: list index out of range
  23. interleaving 2 lists of unequal lengths [duplicate]
  24. Python: filtering lists by indices
  25. How to find the max number(s) in a list with tied numbers
  26. What is the difference between using squared brackets or dot to access a column?
  27. Pandas expand rows from list data available in column
  28. Convert list of strings to dictionary
  29. Pandas – Compute z-score for all columns
  30. How is numpy’s fancy indexing implemented?

Leave a Comment