Numpy Array Get Index Of Conditional Distribution Vs Joint