Данные (число), которые я добавляю в DataFrame с помощью panda

Question

Я построил модель предсказания CNN, чтобы делать предсказания. И я добавляю результат в DataFrame как pi c show. Однако почему в моем столбце прогнозов мои данные заключены в две скобки [[]]? Как от этого избавиться и показать только номер?

test_img = 'C:/Users/User/Desktop/GF_BSIF/Circle_Cropped_test_images/*.jpg'
Test_Path = 'C:/Users/User/Desktop/GF_BSIF/Circle_Cropped_test_images'
Name = []
result = []
for name in os.listdir(Test_Path):
    Name.append(name[0:-5])
for img in glob.glob(test_img):
    prediction = model.predict(prepare(img))
    result.append(prediction)
Temp = {'File Name':Name, 'Prediction':result}
temp = pd.DataFrame(Temp)
temp

So After I change the result.append(prediction) to result.append(prediction[0])

test_img = 'C:/Users/User/Desktop/GF_BSIF/Circle_Cropped_test_images/*.jpg'
Test_Path = 'C:/Users/User/Desktop/GF_BSIF/Circle_Cropped_test_images'
Name = []
result = []
for name in os.listdir(Test_Path):
    Name.append(name[0:-5])
for img in glob.glob(test_img):
    prediction = model.predict(prepare(img))
    result.append(prediction[0])
Temp = {'File Name':Name, 'Prediction':result}
temp = pd.DataFrame(Temp)
temp

The DataFrame shows 1 bracket left. Any way to remove the bracket? введите описание изображения здесь

Answer 1

кажется, что ваш прогноз - это 2d-массив, можете ли вы распечатать тип возвращаемого значения и значение model.predict (). как показано ниже:

print(type(prediction))
print(prediction)