如何用文件名制作熊猫列？答案

【问题标题】：How to make a pandas column out of filenames?如何用文件名制作熊猫列？
【发布时间】：2019-07-13 16:19:00
【问题描述】：

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

【问题讨论】：

请在您的问题中提供更多详细信息stackoverflow.com/help/how-to-ask
通过提供适当的示例在您的问题中添加示例数据和预期输出

【解决方案1】：

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

array = []

然后将这些文件名保存到 pandas.DataFrame 中：

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

【讨论】：

【解决方案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

【讨论】：

亲爱的 tawab，谢谢。我把 file_names = glob.glob('/home/nis/Downloads/trialdata/testset/test-set/*.jpg') 和输出是整个路径位置和文件名。如何仅打印带有扩展名的文件名。