import numpy as np
data_save=np.array([[8,2,3],[4,5,1]])
index_max = np.argmax(data_save, axis=0)
value=data_save[index_max,range(data_save.shape[1])]
print(value)
numpy(二维)数组按列求取最大值
最新推荐文章于 2022-11-14 22:15:19 发布
import numpy as np
data_save=np.array([[8,2,3],[4,5,1]])
index_max = np.argmax(data_save, axis=0)
value=data_save[index_max,range(data_save.shape[1])]
print(value)