【问题标题】:From binary data to integers python [duplicate]从二进制数据到整数python [重复]
【发布时间】:2018-11-02 03:53:13
【问题描述】:

我有一个大小为(N,) 的numpy 向量,其中包含1-5 之间的整数,并且通过使用keras 函数to_categorical,我正在构造大小为(Nx5) 的相应二进制矩阵。例如,如果向量的第一个值为 1,则数组的第一行为(1, 0, 0, 0, 0)。我该怎么做相反的事情?通过有一个值数组(也可以加倍值)来返回具有最高值的索引?有没有可以自动执行的命令?

作为一个例子,我的输入可以是一个带有双精度的二维数组,例如一行可以是[[0.025, 0.022, 0.58, 0.011, 0.22 ]....],结果是该行的[3 ...]

【问题讨论】:

  • 也许您可以发布一个输入示例并包含所需的输出?
  • 我的输入是一个带有双精度的二维数组,例如,一行可以是 [[0.025, 0.022, 0.58, 0.011, 0.22 ]....],结果是该行 [3 . ..].
  • 您绝对应该将该示例添加到您的问题中。你可以看看如何提供minimal reproducible example

标签: python numpy


【解决方案1】:

如果a 是您的数组,请使用:

a.argmax(axis=1)

返回数组每一行最大值的索引。

演示

>>> a = np.array([[5,2,3],[1,3,1]])
>>> a.argmax(axis=1)
[0 1]

【讨论】:

    猜你喜欢
    • 2013-03-28
    • 1970-01-01
    • 2021-10-13
    • 1970-01-01
    • 2011-12-30
    • 2021-12-28
    • 2014-01-15
    • 2014-06-10
    • 2016-01-06
    相关资源
    最近更新 更多