您的位置:首页 > 编程语言 > Python开发

python常用知识点小结(持续更新)

2018-03-09 19:23 417 查看

1,argmax()

(1)针对一维数组:返回数组中最大值元素的索引位置,如下代码所示,输出结果为元素6的索引值:4import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
print(np.argmax(a)) #输出结果为4(2)针对二维数组:参数axis可取(0,1),默认是0,表示数组第几维的最大值。下面分别针对axis=0,axis=1进行举例解释
axis = 0,简单理解,返回结果为:数组中每一列最大值所在“行”索引值。如下代码所示:第一列最大值是9(所在“行”索引值是1),所以针对该列的输出值为1;第二列最大值为7(所在“行”索引值是2),所以针对该列的输出值为2;同理,针对第三列的输出值为2,针对第四列的输出值为1;所以最终的输出值为(1,2,2,1)
import numpy as np
a = np.array([[1, 5, 5, 2],
[9, 6, 2, 8],
[3, 7, 9, 1]])
print(np.argmax(a, axis=0)) #输出结果为(1,2,2,1)
axis = 1,简单理解,返回结果为:数组中每一行最大值所在“列”索引值。如下代码所示:第一行最大值为5(所在列索引值为1),所以针对第一行的输出值为1;同理,针对第二,第三行的输出值分别为:0,2

import numpy as np
a = np.array([[1, 5, 5, 2],
[9, 6, 2, 8],
[3, 7, 9, 1]])
print(np.argmax(a, axis=1)) #输出结果为(1,0,2)
(3)针对三维的数组,推荐一篇博文  http://www.cnblogs.com/zhouyang209117/p/6512302.html
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: