在没有扩展名的目录中获取文件名 - Python

Question

我试图在没有获得扩展名的情况下获取目录中的文件名。使用我当前的代码 -

import os

path = '/Users/vivek/Desktop/buffer/xmlTest/' 
files = os.listdir(path) 
print (files)

我最终得到这样的输出：

['img_8111.jpg', 'img_8120.jpg', 'img_8127.jpg', 'img_8128.jpg', 'img_8129.jpg', 'img_8130.jpg']

但是我希望输出看起来更像这样：

['img_8111', 'img_8120', 'img_8127', 'img_8128', 'img_8129', 'img_8130']

我怎样才能做到这一点？

Answer 1

您可以使用os的拆分文本。

import os

path = '/Users/vivek/Desktop/buffer/xmlTest/' 
files = [os.path.splitext(filename)[0] for filename in os.listdir(path)]
print (files)

只是提醒一下：basename不会为此工作。 basename不会删除扩展名。