Module skm_pyutils.array

Array manipulation.

"""Array manipulation."""
import numpy as np


def non_zero_divide(numerator, denominator):
    """Divide numerator / denominator, but if denominator is zero result is zero"""
    result = np.divide(
        numerator, denominator, out=np.zeros_like(numerator), where=denominator != 0
    )
    return result

Functions

def non_zero_divide(numerator, denominator)

Divide numerator / denominator, but if denominator is zero result is zero

Expand source code

def non_zero_divide(numerator, denominator):
    """Divide numerator / denominator, but if denominator is zero result is zero"""
    result = np.divide(
        numerator, denominator, out=np.zeros_like(numerator), where=denominator != 0
    )
    return result