Python Numpy Array Sets