https://camera-optical-zoom.blogspot.com/2024/02/get-indices-of-numpy-1d-array-where.html