행렬에서 가장 높은 값의 위치를 ​​얻는 방법

Question

행렬이 주어진 경우 행렬에 행을 요청하고 그 행에서 가장 높은 값의 위치를 ​​제공하는 프로그램을 만들어야합니다. 매트릭스와 예

:

1을 반환한다 값이 주어

D = [(0 1 3), 
    (1 0 4), 
    (3 4 0)] 

(인해 행 [2] 높은 값이 그 행에서 1의 위치에있는 4).

지금 내가 노력하고있어 :

def farthest(matrix, row, point): 
    maxdist = 0 
    matrix2 = [] 
    for i in range(0, len(matrix)): 
     if i == int(point): 
      matrix2.append(i) 
     if i != int(point): 
      pass 
     for j in range(0, len(matrix2)): 
      if j < maxdist : 
       pass 
      if j > maxdist: 
       maxdist = maxdist + j 
    print(matrix2) 
    print(maxdist) 
    return matrix2 

내가 루프
내가 얻을 전류 출력을 사용하여 해결책을 찾아야한다 [2] 및 [0]

Answer 1

당신은 NumPy와 사용할 수 있습니다

및 이 경우 1 축당 argmax을 입력하여 행 당 최대 값의 위치를 ​​가져옵니다. 당신

array([2, 2, 1]) 

줄 것이다

import numpy as np 

D = [(0, 1, 3), 
    (1, 0, 4), 
    (3, 4, 0)] 

t = np.array(D).argmax(axis=1) 
t 

은 다음 행이 지수는

t[2] 
1