Необходимо добавить элементы массива в список.Попытка использовать "tolist", "extend" и "append", но безуспешно

Question

Я пытаюсь создать список с image_id и несколькими процентами прогноза. Процент прогнозирования (yhat) находится в массиве np. Я думал, что использование «tolist», затем «extends», а затем «append» должно работать. Я должен делать что-то не так. Большое спасибо за любую помощь!

# Predict the probability across all output classes.
yhat = model.predict(image_array) #the image predictions
yhat.tolist() #change array to list
row = [i[0]] #the image id 
row.extend(yhat) #add list elements to row
result_yhat.append(row) #append row to result_yhat

Результат по-прежнему имеет прогнозные числа в массиве.

result_yhat
[['5', array([6.78813876e-07, 1.14646399e-08, 5.51704140e-08, 
    9.05712838e-08,...

Answer 1

Я думаю, что вы хотите присвоить результат yhat.tolist() переменной и передать его append().

В противном случае вы просто передаете массив np в append().

yhat = yhat.tolist() #change array to list
row = [i[0]] #the image id 
row.extend(yhat) #add list elements to row