https://alezinhacris.blogspot.com/2023/12/choose-elements-from-two-numpy-arrays.html