機械学習を勉強している大学生です。
python,tensorflow,kerasを用いて2クラス分類を行うプログラムを書いています。

分類確率をmodel.predictで出すことはできるのですが、
その確率値がどのデータに対するものなのかを出力する方法がわかりません。
以下は確率値を出すための部分的なプログラムです。

result = model.predict(X_test)

import csv
with open('aaa.csv', 'w', newline='') as csv_file:
    title = ['0', '1']
    writer = csv.writer(csv_file)  
    writer.writerow(title)
    writer.writerows(result)

これを改良したいと思っているのですが、方法がわからずです。
理想としては

ファイル名 0である確率値 1である確率値
1.png 0.9999 0.0001
2.png 0.9999 0.0001
.
.
.
のように出力することです。

よろしくお願いします。