[Get Filtered Positions] Take an array and a filtering function return all the positions and values for which taht function is true

get_filtered_pos.py

from itertools import product
def get_filtered_pos(arr, func):
    """
    Take any ndarray and return the position 
    """
    d_ranges = [range(_) for _ in arr.shape]    
    for j in product(*d_ranges):
        if func(arr[j]):
            yield j, arr[j]