新しいファイルを作成、保存したい

Question

##やりたいこと
input.txt のIMG_◯◯.jpg以下の部分をIMG_〇〇.txtというファイルを作ってそこに書き込みたい。

IMG_◯◯.jpg以下の部分は必要に応じて書き換え作業を行っているがその部分はおそらく大丈夫なので、そこの部分の記述は無視してもらっても大丈夫です。

↓今回書いたコード

import os

INPUT_TXT = "input.txt"
OUTPUT_DIR = "outputs"

HEIGHT = 3744
WIDTH = 5616

CATEGORY_TO_NUM = {
    "b": 0,
    "n": 1,
    "u": 2
}

with open(INPUT_TXT) as f:
    lines = f.readlines()

outputs = {}
current_key = ''

for line in lines:
    if line.startswith("IMG_"):
        outputs[line] = []
        current_key = line
    else:
        splited = line.split(",")
        x = int(splited[0])
        y = int(splited[1])
        w = int(splited[2])
        h = int(splited[3])
        c = CATEGORY_TO_NUM[splited[4].strip()]

        x_center = x + w / 2
        y_center = y + h / 2

        x_center /= WIDTH
        y_center /= HEIGHT
        w /= WIDTH
        h /= HEIGHT

        output_s = f"{c} {x_center} {y_center} {w} {h}"
        outputs[current_key].append(output_s)

for key, value in outputs.items():
    output_fname = f"{os.path.splitext(key)[0]}.txt"
    #output_fname = key.replace("jpg", "txt")
    with open(os.path.join(OUTPUT_DIR, output_fname), mode='w') as f:
        f.write('\n'.join(value))

↓input.txt

IMG_2012-09-26T09-00-01.jpg
4640,3132,17,15,u
4882,3158,16,24,u
707,3273,23,19,u
IMG_2012-09-26T09-00-03.jpg
5218,3156,32,24,b
IMG_2012-09-26T09-00-05.jpg
5528,3106,38,27,b
4139,3164,28,21,b
691,3258,35,19,b
516,3319,21,28,b
5272,3146,31,29,b
IMG_2012-09-26T09-00-07.jpg
5552,3140,29,15,b
1167,3164,40,25,u
IMG_2012-09-26T09-00-09.jpg
1958,2622,20,16,b
543,3316,36,23,b
795,3248,44,23,b
IMG_2012-09-26T09-00-11.jpg
3394,2160,32,22,b
784,3236,20,16,b

#エラーメッセージ

Traceback (most recent call last):
  File "main.py", line 48, in <module>
    with open(os.path.join(OUTPUT_DIR, output_fname), mode='w') as f:
FileNotFoundError: [Errno 2] No such file or directory: 'outputs/IMG_2012-09-26T09-00-01.txt\n'

存在しないファイルに書き込もうとしているためエラーが起こってしまっているのだと思いますが、ファイルが存在しない場合はあたらしくファイルを作って書き込むにはどうしたら良いでしょうか？

Answer 1

FileNotFoundError: [Errno 2] No such file or directory: 'outputs/IMG_2012-09-26T09-00-01.txt\n'

まず、この文字列後尾に入ってる\nを削除する必要があります
次に、この記述だとカレントディレクトリにあるoutputフォルダにファイルを作る、ということになろうかと思いますが、このカレントディレクトリのoutputフォルダというのは存在してるでしょうか。
これが存在してないと、同様のエラーが出ます

Comments

[jagaimo_potato]

記述を忘れていましたがoutputs というファイルは存在します。
おかげさまで\nを削除することができました。ありがとうございました。

Answer 2

17行目と18行目のlineの部分をline.strip()に書き換えることで、作成したいファイルの名前の最後に\nがついてしまうのを改善し、下から二行目の前にos.makedirs(OUTPUT_DIR, exist_ok=True)を追加することでエラーをなくしました。