pythonの配列検索、エラー修正

Question

test6sample.txtからランダムに4組の文章をとってきて
1つ目の文にあり、かつ2つ目の文にも存在するフレーズを出力し次のように表現したいコードを書きたいのですが、59行目でたまにエラーを吐きます。str1[word_line+b+1]が存在しませんと出るのですが58行目でエラーを回避してるつもりができていません。

やりたいこと
2つの文章がこうだとする
1つ目
abcxxxacxxx
2つ目
zzzabczzzzz
理想の出力
[[abc,xxx,a,c,xxx][1,0,1,1,0]]

こんな感じに出力したいのですが、文字列を取得したときに改行も含んでしまうため、コードを書くのにすごく複雑になってしまいました。

# coding: utf-8
import random


for x in range(4):
    line_num = len(open("test6sample.txt").readlines())
    num_1 = 0
    num_2 = 0
    while num_1 == num_2:
        num_1 = random.randint(0,line_num - 1)
        num_2 = random.randint(0,line_num - 1)
    #num_1 = 7
    #num_2 = 8
    f = open("test6sample.txt", "r")
    str1 = f.readlines()[num_1]
    f.close()
    f = open("test6sample.txt", "r")
    str2 = f.readlines()[num_2]
    f.close()

    #単語検出プログラム
    word_lines = 0
    word_len = 0
    word_else = []
    print(len(str1), str1)
    print(len(str2), str2)
    open("test6-2.txt","a").write("[[")
    while word_lines < len(str1):
        word_len = 0
        #print("point1","wl=",word_lines,"wn=",word_len,str1[word_lines])
        for b in range(len(str1)-word_lines):
            word_len += 1
            if str1[word_lines:word_lines+b] in str2 and str1[word_lines:word_lines+b] != "":
                #print("point2","wl=",word_lines,"wn=",word_len,str1[word_lines])
                if str1[word_lines:word_lines+b] in str2 and not(str1[word_lines:word_lines+b+1] in str2):
                    #print("point3","wl=",word_lines,"wn=",word_len,str1[word_lines])
                    open("test6.txt","a").write(str1[word_lines:word_lines+b] + "\n")
                    if str1[word_lines+b] != "\n":
                        #print("point4","wl=",word_lines,"wn=",word_len,str1[word_lines])
                        open("test6-2.txt","a").write(str1[word_lines:word_lines+b]+",")
                        word_else.append(1)
                    else:
                        #print("point5","wl=",word_lines,"wn=",word_len,str1[word_lines])
                        open("test6-2.txt","a").write(str1[word_lines:word_lines+b])
                        word_else.append(1)
                    #print("出力1:                               ",str1[word_lines:word_lines+word_len])
                    word_lines = word_lines + b - 1
                elif str1[word_lines+b] == "\n":
                    open("test6-2.txt","a").write(str1[word_lines:word_lines+b])
                    word_lines = word_lines + b - 1
                    word_else.append(1)
            else:
                #print("point6","wl=",word_lines,"wn=",word_len,str1[word_lines])
                if str1[word_lines:word_lines+b] in str2 and not(str1[word_lines:word_lines+b+1] in str2) and str1[word_lines+b] != "\n":
                    #print("point7","wl=",word_lines,"wn=",word_len,str1[word_lines])
                    open("test6-2.txt","a").write(str1[word_lines+b])
                    #print("出力2:                               ",str1[word_lines+word_len])
                    if str1[word_lines+b] != "\n":
                        if str1[word_lines+b+1] in str2 and str1[word_lines+b] != "\n":
                            #print("point8","wl=",word_lines,"wn=",word_len,str1[word_lines])
                            if str1[word_lines+b+1] != "\n":
                                #print("point9","wl=",word_lines,"wn=",word_len,str1[word_lines])
                                open("test6-2.txt","a").write("@")
                                word_else.append(0)
                            else:
                                #print("point0","wl=",word_lines,"wn=",word_len,str1[word_lines])
                                word_else.append(0)
        word_lines += 1
    open("test6-2.txt","a").write("]")
    open("test6-2.txt","a").write("," + str(word_else))
    open("test6-2.txt","a").write("]\n")    
open("test6-2.txt","a").write("\n")

Answer 1

こんな感じに出力したいのですが、文字列を取得したときに改行も含んでしまうため、コードを書くのにすごく複雑になってしまいました。

• 文字列から改行コードを削除する用途にはstr.rstripが使えます。

with open("test6sample.txt", "r") as f:
    str1 = f.readlines()[num_1].rstrip('\r\n')
with open("test6sample.txt", "r") as f:
    str2 = f.readlines()[num_2].rstrip('\r\n')

• test6sample.txtの内容が１行だけの時に、

while num_1 == num_2:
    num_1 = random.randint(0,line_num - 1)
    num_2 = random.randint(0,line_num - 1)

random.randint(0, line_num - 1)がどちらも0を返すため、while num_1 == num_2:の条件にて無限ループになります。

• while word_lines < len(str1):のループ内の処理が長いので、関数化/クラス化してみてはー。