Question

我有以下数组：

prediction = [0 1 0 0 0 1 0 0 0 0 0 1 1 1 1 1 0 0 1 0]

我正在遍历测试数据集中的图像，我想要的是每张图像，以获得相应的预测。因此，例如，对于image1，预测将为0，对于image2，预测将为1，等等。

我一直在尝试这样的事情，我知道这是错误的，但它可以使您了解我想要的东西：

clf = svm.SVC(kernel='linear', C=100)

for file in glob.glob(test_path + "/*.jpg"):
.
.
.
    prediction = clf.predict(X_test)

    for i in prediction:
        prediction = prediction[i]
        print(prediction)

（我省略了与代码无关的部分，但是如果您需要查看它，我会编辑该帖子并将其添加到其中）

有没有一种方法可以满足我的要求？

谢谢！

Answer 1

您可以这样做：

for index,file in enumerate(glob.glob(test_path + "/*.jpg")):
    prediction[index] #your prediction for the indexth image

这适用于所有迭代：

for i, each in enumerate(iterable):
    print('The'+str(i)+'th element in your iterable is'+str(each))

如何在for循环Python中拆分数组

1 个答案: