使用python将文件列表保存到文本文件中

Question

I would like to get the file name without its extension/suffix with split with specific character.

There are so many jpg files in directory.

for example,

A_list(B)_001.jpg

Stack_overflow_question_0.3.jpg

... and hundreds files in some directory

what I want is to get just the file NAMES without their extensions, like:

A_list(B), 001

Stack_overflow_question, 0.3

but with below code,

import os

path = 'D:\HeadFirstPython\chapter3'
os.chdir(path)

data = open('temp.txt', 'w')

for file in os.listdir(path):
    if file.endswith('.jpg'):
        file = file.split('_')
        print(file, file=data)

data.close()

I got like below result.

['A', 'list(B)', '001.jpg']

['Stack', 'overflow', 'question', '0.3.jpg']

Can this be done with less code?

Thanks and kind regards, Tim

Answer 1

if file.endswith('.jpg'):
        file = file.rsplit('_',1)
        print file[0],
        print file[1].rsplit('.',1)[0]
        print(file, file=data)

Answer 2

import glob
import os

path = 'D:\HeadFirstPython\chapter3'
os.chdir(path)
with open("temp.txt","w") as f: # with automatically closes your files
   my_l = [x.replace(".jpg","").rsplit("_",1) for x in glob.glob("*.jpg")] # list of lists



with open("temp.txt", "w") as f:
    for x in glob.glob("*.jpg"):
        print x.replace(".jpg", "").rsplit("_", 1) # each list 

The output will look like:

s = "Stack_overflow_question_0.3.jpg"

print s.replace(".jpg", "").rsplit("_", 1)
['Stack_overflow_question', '0.3']

To write to txt file without "," :

with open("temp.txt", "w") as f:  # with automatically closes your files
    my_l = [x.replace(".jpg", "").rsplit("_", 1) for x in glob.glob("*.jpg")]
    for l in my_l:
        f.write(str(l).replace(",", ""))

Using "*.jpg" will search for any file ending with jpg . rsplit("_",1) will split on the rightmost _ and using 1 as maxsplit will only split once. We simply replace the extension with str.replace .