获取numpy数组每列最高点行坐标

一，问题描述

获取numpy数组每列最高点行坐标

二，运行环境

python 2/3

三， 需要安装的库

pip install numpy

import numpy as np
a = np.array([
       [8, 1, 0, 8],
       [1, 4, 8, 8],
       [7, 8, 7, 6],
       [8, 7, 6, 5]])
(rowa,cola)=np.where(a==np.max(a,axis=0))  # 每列最大元素的索引
rowa=list(rowa)
cola=list(cola)
row_a=[]
col_a=[]
# 去除重复元素
for i in cola:
  if not i in col_a:
      col_a.append(i)
      row_a.append(rowa[cola.index(i)])
col_a = np.array(col_a)
row_a = np.array(row_a)
# 获取每列最大值对应的行坐标
sort = np.argsort(col_a)
d = row_a[sort]