如何将一列图片名称放入数组的csv文件中？

Question

这是我在这里的第一个问题，如果我做错了，我很抱歉。

我有一个包含图片的文件夹，我将其转换为数组并使用 Python 3 放入一个 csv 文件。它工作正常，但我想在 csv 文件的开头添加图片的名称作为一行。如果有帮助，我也有单独文件中的图片（如标签），我还可以遍历单独的文件夹并将文件夹的名称添加为 csv 文件左侧的一行，但我确实需要所有最后将数组放在同一个 csv 文件中。我怎样才能做到这一点？这是我用来将图像转换为数组并保存为 csv 文件的代码：

import numpy as np
import os
import cv2

IMG_DIR = '../train'

for img in os.listdir(IMG_DIR):
        img_array = cv2.imread(os.path.join(IMG_DIR,img), cv2.IMREAD_GRAYSCALE)
        if img_array is not None:
            img_array = (img_array.flatten())
        img_array  = img_array.reshape(-1, 1).T
        with open('train.csv', 'ab') as f:
            np.savetxt(f, img_array, delimiter=",")
        print(img_array)
            

Answer 1

改进的答案（生成每行包含 filename, pixel_value_array 的 csv 文件）

with open("train2.csv", "w") as f:
    for img_name in os.listdir(IMG_DIR):
        img_array = cv2.imread(os.path.join(IMG_DIR, img_name), cv2.IMREAD_GRAYSCALE)
        f.write(img_name + ",")
        np.savetxt(f, img_array.reshape((1, -1)), delimiter=",")

原答案：先写下所有文件名，再追加数组数据。

images = os.listdir(IMG_DIR)
with open("train.csv", "w") as f:
    f.write("\n".join(images) + "\n")
for img in images:
    img_array = cv2.imread(os.path.join(IMG_DIR, img), cv2.IMREAD_GRAYSCALE)
    img_array = img_array.reshape(-1, 1).T
    with open("train.csv", "ab") as f:
        np.savetxt(f, img_array, delimiter=",")