pythonでのtxtファイルの書き換え

Question

前提・実現したいこと

Frame #:  1						
ID:1	KPHN	(866	800	965	 850)	
ID:2	KPHN	(796	850	851	 904)	
Frame_time:0.100						
Frame #:  2						
ID:1	KPHN	(866	800	967	 851)	
**ID:3**	KPHN	(796	852	851	 906)	
Frame_time:0.133						
Frame #:  3						
ID:1	KPHN	(873	799	976	 851)	
ID:3	KPHN	(796	852	853	 909)	

これを

Frame #:  1						
ID:1	KPHN	(866	800	965	 850)	
ID:2	KPHN	(796	850	851	 904)	
Frame_time:0.100						
Frame #:  2						
ID:1	KPHN	(866	800	967	 851)	
**ID:2**	KPHN	(796	852	851	 906)	
Frame_time:0.133						
Frame #:  3						
ID:1	KPHN	(873	799	976	 851)	
**ID:2** KPHN	(796	852	853	 909)	

ソース（未完成）

f = open('out.csv', 'r', encoding='UTF-8')
i= open('inp.txt','w', encoding='UTF-8')
a="ID:"
for data in f:
    if a in data:
        i.write(data)
    elif a not in data:
        i.write(data)


f.close()

やりたいこと

このような形式で表記されているtxtファイルがあるとして右4つの座標点のでどれかが＋ー5の範囲内ならおなじID に書き直すというプログラムを書きたいです

これはYolov4とDeepsortを用いKPHNという物体をトラッキングしているのですが誤認識などでもともとIDが振られているものに対して新たに新しいIDが振られているのでそれを直したいという感じです。

上の場合でいうとID2とID3は同一物体なのに精度の問題でID3と振られています。

Comments

[TakaiY]

いろいろやってみたようですが、そのときのコードを提示いたでければアドバイスできると思いますよ。
あと、変換した結果としてどのような結果、ファイルとか出力とかを得たいのかも明記するといいと思います。

また、提示されているものは、どう見てもCSVファイルではありません。
なので、CSVファイルだと思って処理しようとするとできないと思います。

Answer 1

作るとしたら、こんな感じでしょうかね。

1. 元ファイルを開く
2. 出力先ファイルを開く
3. 空のID辞書を作る
4. 元ファイルの内容を1行ずつ読み込んで処理
   a 行頭がID:<数> でなければ、そのまま出力
   b 行頭がID:<数> なら()内の4つの数値を取り出し、ID辞書のエントリのすべてについて、4つの数値と比較する。
   1 条件(全て±5以内)に合うものがあれば、ID:<数> の部分を書き換えて出力。
   2 なければ、そのまま出力し、ID辞書に ID:<数>をキーとして4つの数値を格納する。

Comments

[izm2192]

ありがとうございます！！
わからない部分もあるので調べつつ参考にさせていただいてくんでみたいとおもいます！か

[izm2192]

難しいです。。
pythonまだ勉強したてで一応３版まではできたのですがその後の文字列の条件分岐などが理解できずで。
書いている途中のソース書いておきます荷でお力添えお願いしたいです。
f = open('out.txt', 'r', encoding='UTF-8')
i= open('inp.txt','w', encoding='UTF-8')
a=("ID:")
dic = {}
for data in f:
print(data,file=i)

f.close()

[TakaiY]

この質問欄だと、ソースコードの貼り付けに対応してなくて、インデントが反映できないので、質問を修正して貼り付けてみてください。

内容については、確認してコメントしますね。

[TakaiY]

for 文の中でdataに各行が入っていることは、わかっていますよね？

「行頭がxxxなら」という判定ができませんか？
そういう判定をするには正規表現を使うのが一般的です。 正規表現はとっつきにくいですが強力ですのでぜひ使ってみてください。

[izm2192]

今質問のとことにソースを追加してみました！
一応IDという文字があるかどうかを書く行に対してやるというのはできました
次は座標点の比較をしたいのですがどのようにすれば座標点だけを比較対象にできるのかがわからなく詰まってしまいました。。
正規表現について調べてみます

[TakaiY]

正規表現の例としては、こんな感じでしょう。

r'ID:(\d+) +KPHN +((\d+) +(\d+) +(\d+) +(\d+))'

これを、re.match()に与えます。
()が5つ付けてありますが、それぞれについて、結果のgroups()関数であとで参照できるようにしてあります。

Answer 2

ファイルを変数に読み込み、その変数の内容を修正し、それをファイルに書き込む、と言う手順になります
修正は、ファイルに対して、ではなく変数（文字列変数？）に対しての修正となります
それでコードを書いてみてはいかがでしょうか

Comments

[izm2192]

やってみますっ
ありがとうございます！