https://lowcostwallchargeripod.blogspot.com/2023/10/find-indices-of-numpy-array-based-on.html