如何使用numpy计算数组中值的比例?

时间:2017-01-24 19:26:06

标签: python numpy

嗨我有一个巨大的数组,我想找到数组中索引号最高的数字的比例。所以这会回来:

np.array([
  [ [5, 2, 2], [2, 5, 10] ] 
])
array([ 0.5, 0, 0.5 ])

此外,如果最高数字是重复数据,例如[12,25,25],那么我不想忽略它,但在计算比例时将其包含在分母的计数中。

我想输入一个内部数组大小为3的二维数组,并希望找出整个数组中最大值的分布。

1 个答案:

答案 0 :(得分:1)

  • 使用numpy.amax
  • 查找沿零轴的最大值
  • 创建一个布尔数组,其中array == maximum
  • 使用numpy.sum
  • 计算布尔数组沿零轴的所有True值
  • 通过将计数除以原始数组中的行数来查找比例 - array.shape[0]