Finding the index of elements based on a condition using python list comprehension

by

Another way:

>>> [i for i in range(len(a)) if a[i] > 2]
[2, 5]

In general, remember that while find is a ready-cooked function, list comprehensions are a general, and thus very powerful solution. Nothing prevents you from writing a find function in Python and use it later as you wish. I.e.:

>>> def find_indices(lst, condition):
...   return [i for i, elem in enumerate(lst) if condition(elem)]
... 
>>> find_indices(a, lambda e: e > 2)
[2, 5]

Note that I’m using lists here to mimic Matlab. It would be more Pythonic to use generators and iterators.

More Related Contents:

  1. How do I select rows from a DataFrame based on column values?
  2. Flask raises TemplateNotFound error even though template file exists
  3. ‘pip install’ fails for every package (“Could not find a version that satisfies the requirement”) [duplicate]
  4. Configuring Spark to work with Jupyter Notebook and Anaconda
  5. How to set selenium webdriver from headless mode to normal mode within the same session?
  6. Python: Excluding Modules Pyinstaller
  7. How to programmatically set a global (module) variable?
  8. How to create a copy of a python function [duplicate]
  9. Fastest way to populate QTableView from Pandas data frame
  10. Python smtplib proxy support
  11. python: why does replace not work?
  12. Getting “ImportError: DLL load failed: The specified module could not be found” when using cx_Freeze even with tcl86t.dll and tk86t.dll added in
  13. How to set an HTTP proxy in Python 2.7?
  14. Python.h header file missing on Mac OS X 10.6
  15. Can multiprocessing Process class be run from IDLE
  16. Comparing numbers give the wrong result in Python
  17. Row-wise average for a subset of columns with missing values
  18. How to programmatically generate markdown output in Jupyter notebooks?
  19. How can I match an exact word in a string?
  20. PySpark in iPython notebook raises Py4JJavaError when using count() and first()
  21. Class Based Views VS Function Based Views
  22. Make a py2exe exe run without a console?
  23. PyQt: Why does new window close immediately after opening it [duplicate]
  24. PyQt5 button lambda variable becomes boolean [duplicate]
  25. How do I make matplotlib open a box for user comments?
  26. How to run a Jupyter notebook with Python code automatically on a daily basis?
  27. Write OpenCV image in memory to BytesIO or Tempfile
  28. How to create a django ViewFlow process programmatically
  29. numpy faster than numba and cython , how to improve numba code
  30. Move X-Axis label downwards, but not X-Axis Ticks in matplotlib

Leave a Comment