python - 我使用熊猫附加到 DataFrame 的数据（数字）

Question

我建立了一个 CNN 预测模型来进行预测。我将结果附加到 DataFrame 中，如图所示。但是，为什么我的预测列在我的数据周围有 2 个括号 [[]]？如何摆脱它并只显示数字？

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

所以在我将 result.append(prediction) 更改为 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

DataFrame 显示左侧 1 个括号。有什么办法可以去掉支架？

标签： pythonconv-neural-network

Answer 1

看来您的预测是二维数组，您可以打印 model.predict() 的返回类型和值吗？如下所示：

print(type(prediction))
print(prediction)