python - 为数组赋值(Python、Numpy)

Question

我已经初始化了一个数组，其形式为

for x in range(0,21000): 
    if np.amax(probability[x]) > 0.2:
        i = i+1

标签 = np.zeros(shape=(i,130))

现在通过以下 for 循环，我想将值分配给标签的每一行

for x in range(0,21000):
    if np.amax(probability[x]) > 0.2:
        Label[i] = [feature[x],y[x],np.amax(probability)[x]]
    i = i+1

其中feature是一个128*21000的数组。不幸的是，我不断收到以下错误:

Label[i] = [feature[x],y[x],np.amax(probability)[x]]
IndexError: invalid index to scalar variable.

Answer 1

您不需要 np.amax 的单个标量值而是一组值。因此，您需要为调用指定特定的轴，如下所示:

np.amax(probability, axis=1)

您可以查看 axis 的使用示例来自 docs 的论点

或者，如果您确实想使用标量值，请不要使用 [x] 对其进行索引在np.amax(probability)[x]