一,问题描述
获取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]