Iterating over a numpy array

by

I think you’re looking for the ndenumerate.

>>> a =numpy.array([[1,2],[3,4],[5,6]])
>>> for (x,y), value in numpy.ndenumerate(a):
...  print x,y
... 
0 0
0 1
1 0
1 1
2 0
2 1

Regarding the performance. It is a bit slower than a list comprehension. 

X = np.zeros((100, 100, 100))

%timeit list([((i,j,k), X[i,j,k]) for i in range(X.shape[0]) for j in range(X.shape[1]) for k in range(X.shape[2])])
1 loop, best of 3: 376 ms per loop

%timeit list(np.ndenumerate(X))
1 loop, best of 3: 570 ms per loop

If you are worried about the performance you could optimise a bit further by looking at the implementation of ndenumerate, which does 2 things, converting to an array and looping. If you know you have an array, you can call the .coords attribute of the flat iterator. 

a = X.flat
%timeit list([(a.coords, x) for x in a.flat])
1 loop, best of 3: 305 ms per loop

More Related Contents:

  1. Converting between datetime, Timestamp and datetime64
  2. Convert Python sequence to NumPy array, filling missing values
  3. numpy matrix vector multiplication [duplicate]
  4. Python out of memory on large CSV file (numpy)
  5. Conditionally fill column values based on another columns value in pandas
  6. How to convert an array of strings to an array of floats in numpy?
  7. How do I convert a PIL Image into a NumPy array?
  8. How do I remove NaN values from a NumPy array?
  9. NumPy 1.21.2 may not yet support Python 3.10
  10. numpy get index where value is true
  11. Python vectorizing nested for loops
  12. numpy arbitrary precision linear algebra
  13. How to plot data from multiple two column text files with legends in Matplotlib?
  14. np arrays being immutable – “assignment destination is read-only”
  15. PIP Install Numpy throws an error “ascii codec can’t decode byte 0xe2”
  16. vectorize conditional assignment in pandas dataframe
  17. Efficient way to partially read large numpy file?
  18. Pandas: How to create a datetime object from Week and Year?
  19. Binning a column with pandas
  20. Convert numpy type to python
  21. How to update matplotlib’s imshow() window interactively?
  22. Screen Capture with OpenCV and Python-2.7
  23. Find the index of the k smallest values of a numpy array
  24. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  25. Principal component analysis in Python
  26. Numpy – Replace a number with NaN
  27. Numpy: use reshape or newaxis to add dimensions
  28. How to use sklearn fit_transform with pandas and return dataframe instead of numpy array?
  29. numpy array TypeError: only integer scalar arrays can be converted to a scalar index
  30. Efficient dot products of large memory-mapped arrays

Leave a Comment