关于python 3.x：如何将文件名放入os.listdir的数组中


How to put filenames into an array from os.listdir

让我们说你输入：os.listdir(r'filepath')

输出为：['a.txt','b.txt','c.txt','d.txt','e.txt']

你怎么能把文件名['a', 'b', 'c', 'd', 'e']放入pandas数据帧？

使用列表理解与DataFrame构造函数：

1
2
3
4
5
6
7
8
9
L =  ['a.txt','b.txt','c.txt','d.txt','e.txt']
df = pd.DataFrame({'col':[x.split('.')[0] for x in L]})
print (df)
  col
0   a
1   b
2   c
3   d
4   e

谢谢@Joe Halliwell的建议，主要优点是通用解决方案，请检查：

1
df = pd.DataFrame({'col': [os.path.splitext(x)[0] for x in L]})
相关讨论