return vals < radius ** 2

def vectorized_app2_simplified(volume, roi, radius):
    m, n, r = volume.shape
x, y, z = np.ogrid[0:m, 0:n, 0:r] - roi
return x ** 2 + y ** 2 + z ** 2 < radius ** 2
Python векторизация вложенных циклов for