フォルダの中の画像データのファイル名をリストに入れる方法について

Question

filenames = os.listdir(r"AAAA") 
num_files = len(filenames)

t = np.zeros((num_files),dtype='int32') 
for i,filename in zip(range(num_files),filenames):
    a = filename.split("_")[-1]
    t[i] = a

AAAAディレクトリに***_100.jpg, ***_200.jpgのように画像データが保存してあります。
この画像のファイル名のアンダーバー以下の数字をを読み込んで、t[0] = 100, t[1] = 200 のようにリストに保存したいです。上記のようにプログラムを書いてみましたが、うまくいきませんでした。よろしくお願いします。

Answer 1

filename.split("_")[-1]は"100.jpg"のような結果が得られるので、拡張子を取り除く必要があります。また、intに型変換する必要があります。

また、こういった処理では最初からnumpy配列にしてもパフォーマンス上のメリットはほぼなく、コードが冗長になるだけです。

numpy配列が必要な場合でも、とりあえず扱いやすいリストで処理し、あとからnumpy配列にすればよいかと。

filenames = os.listdir(r"AAAA") 

lst = []
for filename in filenames:
    a = int(filename.split("_")[-1].split(".")[0])
    lst.append(a)

t = np.array(lst, dtype="int32")

Comments

[Kohei_KESE]

回答ありがとうございます。
追加で申し訳ないのですが、数字がfloat型を設定する時はどうすればいいか教えて頂けませんでしょうか？
よろしくお願いします。

[hayataka2049]

それだと同じ方法ではできませんね
拡張子はすべて.jpgだけですか？

[Kohei_KESE]

.jpgだけです。

[hayataka2049]

では、filename.split("_")[-1].replace(".jpg", "")のようにしてみてください
今回はこれで良いと思いますが、もう少し複雑になってきて文字列操作メソッドで対応するのが辛くなってきたら、正規表現の活用も検討してください

[Kohei_KESE]

ありがとうございます。無事実装することができました。
文字列操作はUNIXと違い直観的ではなく苦労しているので、正規表現での記述を覚えた方が今後楽ができそうですね。