文字列を判定し、前行と比較して値を変更する

Question

(file1)
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
GLY B   3
GLY B   3
GLY B   3
GLY B   3
GLY B  10　　　# 1列めの文字列は同じだが、3列めの数字が2以上増えている。
GLY B  10
GLY B  10
GLY B  10
TRP B  11　　　　
TRP B  11
TRP B  11
LEU B   6
LEU B   6
LEU B   6
LEU B   6
ILE B  13    
ILE B  13

(file2)
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
GLY B   3
GLY B   3
GLY B   3
GLY B   3
GLY B   4
GLY B   4
GLY B   4
GLY B   4
TRP B   5
TRP B   5
TRP B   5
LEU B   6
LEU B   6
LEU B   6
LEU B   6
ILE B   7
ILE B   7

file1をfile2のように書き換えたいです。

具体的には
・3列めの数字が前の行より2以上増えているとき、（その前の行の3列めの数＋１）に変更する。
・1列めの文字列が同じ行では3列めの数字を同じにする。
ただし、file1で1列めの文字列が同じでも、3列めの数字が2以上増えているとき、（その前の行の3列めの数＋１）に変更する。

file1を2行目から読み込んだfile1_2を作成し、1行ずつ比較していたのですが、3列目の数字を変えても、その次の行の数字を前行に揃えることができず、このやり方では無理と判断しました。

path = 'file1'

with open(path) as f:    #file1の2行目からを取り出したfile1_2を作成
   next(f)
   for line in f:
       fileobj = open("file1_2","a")
       print >> fileobj, line.rstrip()

path = 'file1'
path2 = 'file1_2'
f = open(path).readlines()
f2 = open(path2).readlines()

for (line, line2) in zip(f, f2):
  line_res = line[0:3]
  line_num = line[7:9]
  line2_res = line2[0:3]
  line2_num = line2[7:9] 
  if (line_res == line2_res) and (line_num == line2_num):　　
　#file1とfile1_2の1列目と3列目が同じならば、file1をそのまま出力
      fileobj = open("file2","a")
      print >> fileobj, line.rstrip()
  elif (line_res != line2_res) and ((int(line_num) + 1) == (int(line2_num))): 
  #file1とfile1_2の1列目が違うかつ3列目が1増えているならば、file1をそのまま出力
      fileobj = open("file2","a")
      print >> fileobj, line.rstrip()
  elif (line_res != line2_res) and ((int(line_num) + 1) != (int(line2_num))):　　
　#file1とfile1_2の1列目が違うかつ3列目が2以上増えているとき、3列目をfile1の3列目の数字＋１にして出力
      line_num_1 = ((int(line_num)) + 1)
      line_num_2 = (str(line_num_1)).rjust(1)
      tmp = line[:30] + line_num_2 + line[33:]
      fileobj = open("file2","a")
      print >> fileobj, tmp.rstrip()    

(file1)　　　　　　　(file1_2)
PRO B   2          PRO B   2
PRO B   2          PRO B   2
PRO B   2          PRO B   2
PRO B   2          PRO B   2
PRO B   2          PRO B   2
PRO B   2          GLY B   3
GLY B   3          GLY B   3
GLY B   3          GLY B   3
GLY B   3          GLY B   3
GLY B   3          GLY B  10
GLY B  10          GLY B  10  　
GLY B  10          GLY B  10
GLY B  10          GLY B  10
GLY B  10          TRP B  11
TRP B  11          TRP B  11　　　　    
TRP B  11          TRP B  11
TRP B  11          LEU B   6
LEU B   6          LEU B   6
LEU B   6          LEU B   6
LEU B   6          LEU B   6
LEU B   6          ILE B  13
ILE B  13          ILE B  13
ILE B  13

しかし、この方法では

(file2)
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
PRO B   2
GLY B   3
GLY B   3
GLY B   3
GLY B   3
GLY B   4
GLY B  10
GLY B  10
GLY B  10
TRP B  11　　　　
TRP B  11
TRP B  11
LEU B   6
LEU B   6
LEU B   6
LEU B   6
ILE B   7    
ILE B  13

となってしまいます。
前の行の変更を反映して次の行に移るようにするにはどうすればよいのでしょうか。
それとも他にやりようがあるのでしょうか。

Comments

[dice142]

コードをご提示ください。

[退会済みユーザー]

追加しました。ご確認お願いします。

Answer 1

pandas 使って行うと良いのではないでしょうか。

import pandas as pd
import numpy as np

df = pd.read_csv('file1', header=None, sep=' ', skipinitialspace=True)
df[2] = df[2].astype('int')
df[2] = (df[2].diff() != 0).cumsum() + df.iloc[0,2] - 1
print(df)
np.savetxt('file2', df.values, fmt='%s %s %3d')

Comments

[退会済みユーザー]

回答ありがとうございます！
数字は期待通りに変更されました。
しかし、pandasで処理してしまうと、列間の空白が変わってしまうので、そこが問題となっています。

[退会済みユーザー]

なんとかなりそうです。
ありがとうございます。

[magichan]

どもです。
一応、新しい質問の方に回等をしておきました。（期待した答えではないかと思いますが）

[magichan]

おっ、コメントが入れ違いになってしまいました

[退会済みユーザー]

ちなみに、np.savetxtで列をタブ区切りはできないのでしょうか。
np.savetxt('file2', df.values, delimiter="\t", fmt='%s %s %3d')
としても、区切りが空白一つのままになっています。

[magichan]

その場合は fmt側にて指定します。TABは'\t'ですので
np.savetxt('file2', df.values, fmt='%s\t%s\t%3d')
でしょうか

[退会済みユーザー]

できました！
ありがとうございます！！