如何用文件名做成熊猫列？

Question

我有N张图像。我想创建熊猫数据框，并将图像的所有文件名放在这些列中，该怎么做？ 我需要带有标题的列 “文件名”。 a.jpg b.jpg

Answer 1

创建一个数组，在其中添加文件名。

array = []

然后将这些文件名另存为pandas.DataFrame：

df = pd.DataFrame(array, index=False, columns=["filenames"])

Answer 2

import glob
import pandas as pd
# fetch all images from your directory 
# I am assuming .png is the extension of images 
file_names = glob.glob('./*.png') # it will give list of file_names ['a.png','b.png']

#if absolute path is like ["C://abc/img.png","C://abc//img2.png]
images = [i.split("/")[-1]for i in file_names]

#if absolute path is like ["C:\\abc\img.png","C:\\abc\\img2.png]
images = [i.split("\\")[-1]for i in file_names]



df = pd.DataFrame(images,columns=['column_name'])
print(df)

  column_name
0   a.png
1   b.png