p, b = np.broadcast_arrays(p, b)