Как разбить массив в цикле для Python

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)

(я пропустил ту часть кода, которая не имеет значения, но если вам нужно ее увидеть, я отредактирую пост и добавлю его)

IsЕсть ли способ сделать то, что я спрашиваю?

Заранее спасибо!

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))