On 9/8/2010 6:38 AM, John Reid wrote: > def safe_x_log_x(x): > "@return: x log(x) but replaces -inf with 0." > l = np.log(x) > result = x * l > result[np.isneginf(l)] = 0. > return result
Assuming x is known to contain nonnegative floats: def safe_x_log_x(x): x = np.asarray(x) l = np.zeros_like(x) l[x>0] = np.log(x[x>0]) return x * l hth, Alan Isaac _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion