entfernen Erweiterung in Liste Python

Question

Ich habe den folgenden Code, wodurch ich alle Dateien in einem bestimmten Ordner (dieser Ordner enthält. TIF-Dateien) suchen möchte, dann speichern Sie diese Namen in einer TXT-Datei, die soweit funktioniert. Ich möchte jedoch nicht die Erweiterungen von allem, was in dem Ordner wäre.

Nach Kommentaren, vor allem dank ShadowRanger, änderte ich es zu den folgenden, die den gleichen Effekt produziert. Vielen Dank an alle !:

import os 

f = [e.name for e in os.scandir(r'C:\\Users\\user\\Desktop\\FOLDER') if e.is_file()] 
print('Saving file...') 

def save_to_file(text): 
    for name in f: 
     with open('C:\\Users\\user\\Desktop\\FOLDER\\test5.txt', mode='wt', encoding='utf-8') as myfile: 
      myfile.write('\n'.join(text)) 
      myfile.write('\n') 

save_to_file(f) 

Answer 1

Entfernen Sie die Erweiterung in einem Generatorausdruck, wenn Sie die Dateinamen mit os.path.splitext erhalten. Wechsel:

f.extend(filenames) 

zu:

f.extend(os.path.splitext(name)[0] for name in filenames) 

Update: Es sieht aus wie Sie os.walk sind missbrauchen Datei aus einem Verzeichnis-Trennung zu tun (da Sie break in der ersten Schleife, so bist du nicht überhaupt einen Baum zu gehen), wenn es direktere Möglichkeiten gibt, dies zu tun. Sie können alle diese ersetzen:

f = [] 
b = os.walk('C:\\Users\\username\\Desktop\\FOLDER') 
for(dirpath, dirnames, filenames) in b: 
    f.extend(filenames) 
    break 

auf Python 3.5 oder höher (oder ein beliebige Version von Python nach the third party scandir package mit scandir.scandir statt os.scandir installieren), die mit den prägnanten und etwas schneller:

f = [os.path.splitext(e.name)[0] for e in os.scandir(r'C:\Users\username\Desktop\FOLDER') 
    if e.is_file()] 

der beschreibt, was Sie wirklich wollen, und indem Sie os.scandir verwenden, garantiert, dass Sie nicht einzeln stat jeden Eintrag in der Verzeichnisliste, um zu prüfen, ob es eine Datei ist.

Wenn Sie auf älteren Python sind, und das scandir Paket nicht verwenden können, um die Kosten für einen stat (die os.walk ohnehin auf älteren Python zahlen würde), die Sie tun können:

f = [os.path.splitext(x)[0] for x in os.listdir(r'C:\Users\username\Desktop\FOLDER') 
    if os.path.isfile(os.path.join(r'C:\Users\username\Desktop\FOLDER', x))] 

Answer 2

diese Zeile ersetzen:

f.extend(filenames) 

mit etwas wie folgt aus:

filenames_stripped = [os.path.splitext(file)[0] for file in filenames] 
f.extend(filenames_stripped) 

oder noch einfacher:

f.extend([os.path.splitext(file)[0] for file in filenames] 

Weitere Informationen über was os.path.splitext aber es essen teilt den Dateinamen in zwei Teile auf. Die erste ist die Wurzel, die du willst.