フォルダ内の名前をリストとして読み込み、編集する

Question

指定したパスのフォルダ内のファイル名を編集したく、質問させていただきました。
現在、namesというリストの中にファイル名が格納されておりますが、ファイル名が長いため
一部分のみを抜き出したリストとしたいです。
split()を使ってみましたが、うまくいきませんでした

import csv
import os

location = '/.dir/'
names = os.listdir(path=location)　#ここでファイル一覧をnamesというリストとして定義
print(names)

['aaa_bbb_1.0.csv','aaa_bbb_1.1.csv','aaa_bbb_1.2.csv'・・・・]

aaa_bbbはどのファイルも同じ文字となっており、数字だけ抜き出すような
['1.0','1.1','1.2'・・・・]このようなリストを定義したいです。
初歩的な質問となりましたが、どうぞよろしくお願い致します。

Answer 1

では、ちょっと違うやり方で。

import re
lst = ['aaa_bbb_1.0.csv','aaa_bbb_1.1.csv','aaa_bbb_1.2.csv']
lst = [re.search('([\d.]+).csv',t).group(1) for t in lst]
print(lst)
# ['1.0', '1.1', '1.2']

Comments

[K.ooo]

ご丁寧にありがとうございました。
出来るようになりました。

Answer 2

こんなやり方もありますよ！

lst = ['aaa_bbb_1.0.csv','aaa_bbb_1.1.csv','aaa_bbb_1.2.csv']
lst = [t.replace('.csv', '') for t in [s.replace('aaa_bbb_', '') for s in lst]]
print(lst)
# ['1.0', '1.1', '1.2']

Answer 3

lst = ['aaa_bbb_1.0.csv','aaa_bbb_1.1.csv','aaa_bbb_1.2.csv']
lst = list(map(lambda x:x[8:-4], lst))
print(lst)
# ['1.0', '1.1', '1.2']
```Pythonには便利なmap関数があります。

Comments

[quickquip]

今のPythonなら、for文より list(map(lambda の方が遅いかと思います。

[ForestSeo]

そうなんですか！
初めて知りました。