Python3 複数行 リスト インデックス 変数代入

Question

前提・実現したいこと

複数行リストのインデックスに変数を代入して正常に動作させたい。

発生している問題・エラーメッセージ

a: あたりくじ
b: リストの複数行の数
c: リストの中身
result: それぞれのリストの中にあるあたりくじ

該当のソースコード

a = ['1', '2', '3', '4', '5', '6']

b = 3

c = [['1', '5', '4', '2', '3', '6'],
['9', '6', '2', '7', '1', '5'],
['32', '9', '87', '33', '41', '60']]

result_1 = [i for i in set(a) if i in c[0]]
result_2 = [i for i in set(a) if i in c[1]]
result_3 = [i for i in set(a) if i in c[2]]

print(result_1)
print(result_2)
print(result_3)

試したこと

コードの繰り返しのため、for文を利用して簡素かしたい。

しかし、c[0],c[1],c[2]に変数を代入する方法がわかない。

result = []
for i in range(b):
if j in c[i]:
result.append(j)
print(result)

正常に動作しない。

ご教授お願い致します。

Answer 1

提示コードの[i for i in set(a) if i in one]をそのまま使い、以下のようにすればよいと思います。

a = ['1', '2', '3', '4', '5', '6'] 
c = [['1', '5', '4', '2', '3', '6'],
['9', '6', '2', '7', '1', '5'],
['32', '9', '87', '33', '41', '60']] 

results = []
for one in c:
    results.append([i for i in set(a) if i in one])
print(results)

# ２重でも可
results = [[i for i in set(a) if i in one] for one in c]
print(results)

Comments

[python3_beginer]

お返事頂きありがとうございます。
提示して頂いたコードをみて、おもわず「なるほど～！」とスッキリしました！
リスト内包表記まで追記して頂きありがとうございました。

複数行あるリストc　の中身を繰り返し処理をしたいので、項数など考慮せずに
for one in c:
と書けばいいんですね！

昨晩から何度も考えていたので、解決してとても嬉しいです！ありがとうございました！

Answer 2

こんなのでどうでしょう。多重のリスト内包表記とアンパック代入を利用しています。

result_1, result_2, result_3 = [[i for i in set(a) if i in line] 
                                for line in c]

Comments

[python3_beginer]

お返事頂きありがとうございます。

for line in c の全体を繰り返す発想が抜けておりました。
また、このようなスマートな書き方も目からうろこでした。

ご指導して頂きありがとうございました。