Присвоить значение массиву (Python, Numpy)

Question

Я инициализировал массив в виде

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

Label = np.zeros(shape=(i,130))

и теперь со следующим для цикла я хотел бы присвоить значение каждой строке метки

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

Где особенность - массив 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 из документов

Или, если вы действительно хотите использовать скалярное значение, не индексируйте его, используя [x] в np.amax(probability)[x]