Numpy Array Get Index Of Specific Learning