Вам необходимо указать ось при использовании медианы, макс и мин:
import numpy as np
def findoutlier(lis):
omaxs = np.absolute(np.max(lis, axis=1) - np.median(lis, axis=1))
omins = np.absolute(np.min(lis, axis=1) - np.median(lis, axis=1))
return [np.argmax(l) if omax > omin else np.argmin(l) for omax, omin, l in zip(omaxs, omins, lis)]
def main():
mat_x = np.array([[10, 3, 2], [1, 2, 6]])
print(findoutlier(mat_x))
three_mat_x = np.array([[1, 10, 2, 8, 5], [2, 7, 3, 9, 11], [19, 2, 1, 1, 5]])
print(findoutlier(three_mat_x))
выход
[0, 2]
[1, 0, 0]
UPDATE
Как уже упоминалось @ user3483203, вы можете использовать numpy.where :
import numpy as np
def findoutlier(lis):
omaxs = np.absolute(np.max(lis, axis=1) - np.median(lis, axis=1))
omins = np.absolute(np.min(lis, axis=1) - np.median(lis, axis=1))
return np.where(omaxs > omins, np.argmax(lis, axis=1), np.argmin(lis, axis=1))
def main():
mat_x = np.array([[10, 3, 2], [1, 2, 6]])
print(findoutlier(mat_x))
three_mat_x = np.array([[1, 10, 2, 8, 5], [2, 7, 3, 9, 11], [19, 2, 1, 1, 5]])
print(findoutlier(three_mat_x))
main()
выход
[0 2]
[1 0 0]