File size: 136 Bytes
ff6d288
 
 
 
 
1
2
3
4
5
import numpy as np

def circle_mask(xv, yv, radius):
    dist_from_center = np.sqrt(xv**2 + yv**2)
    return dist_from_center <= radius