テキストファイル1に入っている単語からテキストファイル2に入っている単語の行を抜き出す方法

Question

やりたいこと

　やりたいことは、テキストファイルが2つあり1つ目のテキストファイルには単語が沢山入っていて、2つ目のテキストファイルには沢山の単語とその後ろにスペース区切りで数字があります。テキストファイル１の単語からテキストファイル２の単語を見つけ、その後ろの数字が何回出てきたのかを数えたいです。

具体例

テキストファイル１
プリン
かぶ
美味しい
ビール
いまいち
マック
けんだま

テキストファイル２
目　３
プリン　４
一万円　５
ポテト　２
ビール　２
カレンダー　５
けんだま　４

この場合に、テキストファイル１からテキストファイル２の単語を探し出し、プリンとビールとけんだまが一致して、その後ろの数字の４が２回、２が１回というように出力をしたいです。

コード

test_data = open("text1.txt", "r")
mihon_data = open("text2.txt", "r")


lines = test_data.readlines()
kekka = mihon_data.readlines()


for line1 in lines:
    for line2 in kekka:
        ##何らかの処理

test_data.close()
mihon_data.close()

text1.txtには例のテキストファイル１のようなテキストが入っていて、text2.txtには例のテキストファイル２のようなテキストが入っています。
テキストファイルを開いて２つを１行ずつ読みこみ処理を行えばいいのかと思い、ここまでのコードは書いたのですがわかりません。

よろしければ教えていただけないでしょうか。

Answer 1

text2.txtの中身をカンマ区切り、数字を半角で作成しています。
又、Windows環境なのでフルパス指定、ファイル読み込みにencoding="utf-8"を入れていますがご了承願います。

text2.txt
---
目,3
プリン,4
一万円,5
ポテト,2
ビール,2
カレンダー,5
けんだま,4 
---

import os
# カレントパス
CurrentDir=os.path.dirname(os.path.abspath(__file__)) + "\"

test_data = open(CurrentDir + "text1.txt", "r",encoding="utf-8")
mihon_data = open(CurrentDir + "text2.txt", "r",encoding="utf-8")

lines = test_data.readlines()
kekka = mihon_data.readlines()

count_dict={}   # 結果出力辞書 key:検知数値,value:回数

for line1 in lines:
    line1=line1.replace("\n","")

    for line2 in kekka:
        line2 = line2.replace("\n","")
        
        # count_dict辞書になければvalue:1 存在する場合はインクリメント
        if line1 == line2.split(",")[0]:
            item_count = line2.split(",")[1]
            count_dict[str(item_count)] = 1 if item_count not in count_dict else count_dict[str(item_count)] + 1

[(print(key + " が " + str(count_dict[key]) + "回")) for key in count_dict]

test_data.close()
mihon_data.close()

■結果
4 が 2回
2 が 1回

Answer 2

ファイル１をリストに、ファイル２を辞書に読み込めば、あとはCounterで数え上げることができます。
ファイル読込、ファイル２の全角数字を半角に変換などはご自身で実装ください。

from collections import Counter

words = ['プリン','かぶ','美味しい','ビール','いまいち','マック','けんだま']
num_dict = {'目':3,'プリン':4,'一万円':5,'ポテト':2,'ビール':2,'カレンダー':5,'けんだま':4}
nums = []
for word in words:
    if word in num_dict:
        nums.append(num_dict[word])

counter = Counter(nums)
print(counter) # Counter({4: 2, 2: 1})

Answer 3

まず単語と数字をマップして、それから一行ずつ見ていけば良いのでは。

from collections import Counter


word_to_code = {}
with open('text2.txt') as fin:
    for line in map(str.rstrip, fin):
        k, v = line.split()
        word_to_code[k] = int(v)
        
counter = Counter()
with open('text1.txt') as fin:
    for line in map(str.rstrip, fin):
        if line not in word_to_code:
            continue
        
        k = word_to_code[line]
        counter[k] += 1
        
print(counter)

実行結果 Wandbox

Counter({4: 2, 2: 1})

丸投げな感が否めないので、細かな解説はしません。

Comments

[nanu1129]

自分の勉強不足で丸投げになってしまっているのは本当に申し訳なく思います。
にもかかわらず、迅速で的確な回答をありがとうございます。
本当に感謝しています。

[LouiS0616]

低評価を入れた方へ

もちろん評価自体は自由にしていただいて構わないのですが、私にはいまいち低評価の原因が掴めません。
回答に不足あるいは間違いがあるならば、コメントで事由を併せてご指摘ください。

Answer 4

data_1.txt, data_2.txt data.py で書いてみました。
実行結果

data_1.txt

プリン
かぶ
美味しい
ビール
いまいち
マック
けんだま

data_2.txt

目 ３
プリン ４
一万円 ５
ポテト ２
ビール ２
カレンダー ５
けんだま ４

data.py

from collections import Counter

with open('data_2.txt') as fin:
    word_to_num = {ary[0] : int(ary[1]) for ary in [line.split() for line in fin.readlines()] if len(ary) == 2}

with open('data_1.txt') as fin:
    counts = Counter(([word_to_num.get(line.rstrip()) for line in fin.readlines()]))

[print("{0} は {1}回です".format(k, v)) for k, v in counts.items() if k]