Python 効率の良い複数リストの比較方法について

Question

■やりたいこと
比較元の２つのリストと比較先の２つのリストと一致したインデックスを導き
そのインデックスの要素番号から別なリストのデータを抽出する

■試したこと

all_list_1 = [0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]
all_list_2 = [1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4]
all_list_3 = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
filter_list_1 =[0, 0, 1, 1, 2, 2, 2]
filter_list_2 =[2, 3, 4, 3, 1, 3, 4]

result_list = []

for fildata1, fildata2 in zip(filter_list_1, filter_list_2):
    i = 0
    for alldata1, alldata2 in zip(all_list_1, all_list_2):
        if fildata1 == alldata1 and fildata2 == alldata2:
            result_list.append(all_list_3[i])
        i += 1
print(result_list)
#　結果 ['B', 'C', 'H', 'G', 'I', 'K', 'L']

■知りたいこと
上記のコードよりスマートな方法があるか、ご教授お願い致します。
※実行環境はPyhotn2.4.3になります。

Answer 1

とりあえずfilter値をtupleのsetにして、元データからinで抽出すると二重ループがなくなりすっきりするかと思います。

all_list_1 = [0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]
all_list_2 = [1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4]
all_list_3 = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
filter_list_1 =[0, 0, 1, 1, 2, 2, 2]
filter_list_2 =[2, 3, 4, 3, 1, 3, 4]

filter_set = set((f1,f2) for f1, f2 in zip(filter_list_1, filter_list_2))
result_list = [all_list_3[i] for i, v in enumerate(zip(all_list_1, all_list_2)) if (v[0],v[1]) in filter_set]

print(result_list) # ['B', 'C', 'G', 'H', 'I', 'K', 'L']

Answer 2

見栄えを簡単にすることぐらいしかできません。

python2.7.5では以下で動きました。

all_list_1 = [0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]
all_list_2 = [1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4]
all_list_3 = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L"]
filter_list_1 =[0, 0, 1, 1, 2, 2, 2]
filter_list_2 =[2, 3, 4, 3, 1, 3, 4]

for fil_tuple in zip(filter_list_1, filter_list_2):
    for all_tuple, alldata3 in zip(zip(all_list_1, all_list_2), all_list_3):
        if fil_tuple == all_tuple:
            result_list.append(alldata3)

print(result_list)

Answer 3

スマートかどうかは判りませんが、以下の様な方法も考えられます。

lst_all = zip(all_list_1, all_list_2)
lst_filter = zip(filter_list_1, filter_list_2)
result_list = [all_list_3[i] for i, l in enumerate(lst_all) if l in lst_filter]
print(result_list)

#
['B', 'C', 'G', 'H', 'I', 'K', 'L']

※ 結果が異なるのは zip(all_list_1, all_list_2) 側から走査しているためです