前提・実現したいこと
プロググラム初心者です。
pythonで二次元配列の対応付け計算を行いたいです。
文書Aと文書Bを文ごとに二次元配列にいれ、要素ごとに計算し、最大の値の数値とその文書を抜き取りたいです。
すみませんが、わかりやすく教えていただけると幸いです。
計算方法はDice係数を使用しようと考えています。
***********
具体的に説明すると
文書Aの一列目の文a1と文書Bの一列目の文b1の要素数の割合を計算。
文書Aの一列目の文a1と文書Bの二列目の文b2の要素数の割合を計算。
・・・
文書Aの一列目の文a1と文書Bの最後の文bmの要素数の割合を計算。
割合が最大値の文の組み合わせと割合値を出力。
・・・
文書Aの二列の文a2と文書Bの一列目の文b1の要素数の割合を計算。
・・・・
文章Aの最後の文anと文書Bの最後の文bmの要素数の割合を計算。
割合が最大値の文の組み合わせと割合値を出力。
理想の結果
a1とbm(最大値の文書) 数値
a2とbm(最大値の文書) 数値
・・・・
anとbm(最大値の文書) 数値
***********
わかっていないところ
一文ずつ対応を求めて計算していく方法がわからないです。
二重ループを使い、最大値を更新していく気がするのですが、、
該当のソースコード
python
1#文章を二次元リストに追加 2with open("dataA.txt") as f: #文書A 3 A = [line.split() for line in f] 4with open("dataB.txt") as f: #文書B 5 B = [line.split() for line in f] 6 7#Dice係数の計算アルゴリズム 8def dice_similarity_coefficient(list_a,list_b): 9 #集合Aと集合Bの積集合(set型)を作成 10 set_intersection = set.intersection(set(list_a), set(list_b)) 11 #集合Aと集合Bの積集合の要素数を取得 12 num_intersection = len(set_intersection) 13 14 #集合Aの要素数を取得 15 num_listA = len(list_a) 16 #集合Bの要素数を取得 17 num_listB = len(list_b) 18 19 #定義式に従い,Dice係数を算出 20 try: 21 return float(2.0 * num_intersection) / (num_listA + num_listB) 22 except ZeroDivisionError: 23 return 1.0 24 25if __name__== '__main__': 26 list_a = A #集合A 27 list_b = B #集合B 28 29 dsc = dice_similarity_coefficient(list_a,list_b) #Dice係数を計算 30
回答1件
あなたの回答
tips
プレビュー