读取一个文件中的行，然后查找另一个txt fi中列出的所有以4个字母开头的字符串

Question

我有2个txt文件（a和b）。在

文件_a.txt包含一个4个字母组合的长列表（每行一个组合）：

aaaa
bcsg
aacd
gdee
aadw
hwer
etc.

文件_b.txt包含不同长度的字母组合列表（有些带有空格）：

^{pr2}$

我正在寻找一个允许我执行以下操作的python脚本：

1. 逐行读取文件_a.txt
2. 取每个4个字母的组合（例如aaai）
3. 阅读文件_b.txt，找到以4个字母组合开头的所有不同长度的字母组合（例如aaaibjkes，aaailoiersaaageehikjaaa，aaailoiuwegoiglkjaaaike等）
4. 将每次搜索的结果打印在一个以4个字母组合命名的单独txt文件中。在

文件aaai.txt文件公司名称：

aaaibjkes 
aaailoiersaaageehikjaaa
aaailoiuwegoiglkjaaake
etc.

文件密件索引.txt公司名称：

bcspwiopiejowih
bcsiweyoieotpwe
etc.

对不起，我是个新手。有人能给我指一下正确的方向吗。到目前为止，我只有：

#I presume I will have to use regex at some point
import re

file1 = open('file_a.txt', 'r').readlines()
file2 = open('file_b.txt', 'r').readlines()

#Should I look into findall()?

Answer 1

试试这个：

f1 = open("a.txt","r").readlines()
f2 = open("b.txt","r").readlines()
file1 = [word.replace("\n","") for word in f1]
file2 = [word.replace("\n","") for word in f2]

data = []
data_dict ={}
for short_word in file1:
    data += ([[short_word,w] for w in file2 if w.startswith(short_word)])

for single_data in data:
    if single_data[0] in data_dict:
        data_dict[single_data[0]].append(single_data[1])
    else:
        data_dict[single_data[0]]=[single_data[1]]

for key,val in data_dict.iteritems():
    open(key+".txt","w").writelines("\n".join(val))
    print(key + ".txt created")