複数のリストから重複しているものを元に必要なものだけを抽出したリストを作りたい

Question

前提・実現したいこと

初質問です、至らぬ点があったらすみません。

list1 = [a,b,c]
list2 = [a,d,e]
list3 = [x,y,z]

上記のような3つのリストがあったら
list4 = [a,b,d]
list5 = [x,y]

以上のように、他のリストと重複しているものがあるときはまとめた上で必要な要素だけを抽出し、
他のリストと重複していないときは必要な要素だけを抽出したリストをそれぞれ作りたいです。

試したこと

for分を用いて試行錯誤したのですが自分ではうまくできませんでした。
よろしければ教えていただきたいです。

Comments

[meg_]

> 必要な要素だけを抽出し
"必要な要素"って何ですか？

[hinaseen]

上の例だと他のリストと重複している部分がなかったらlist3からは[x,y]だけを取り出したいということです

[meg_]

> 上の例だと他のリストと重複している部分がなかったらlist3からは[x,y]だけを取り出したいということです
何故zを除外したのですか？どんなロジックなのですか？list4についてもロジックが全く分かりません。

[hinaseen]

zは必要のない要素だったので削除しました。それぞれのリストに必要のない要素が入っている状況です。
list4に関してはaという共通した要素を取り出してb,cと必要な要素を付けます。

[meg_]

個別の条件があるんですね。質問に書いてないので意味不明な質問に見えました。
もっと汎用的な関数でも作りたいのかと思いましたが違うということで理解しました。

> for分を用いて試行錯誤したのですが自分ではうまくできませんでした。
途中でも良いのでそのコードを載せないと丸投げに見えてしまいます。

Answer 1

他のリストと重複しているものがあるときはまとめた上で必要な要素だけを抽出し、
他のリストと重複していないときは必要な要素だけを抽出したリストをそれぞれ作りたいです。

必要な要素 が 各リストの最初の２つ を意味し
他のリストと重複 が 他のリストと必要な部分で重複 を意味する場合

期待しているコードは

list1 = ['a', 'b', 'c']
list2 = ['a', 'd', 'e']
list3 = ['x', 'y', 'z']
list4 = ['b', 'f', 'g']

lst = [ls[:2] for ls in (list1, list2, list3, list4)]
ovlp = [(i,k) for i in range(len(lst)) for k in range(i+1, len(lst)) if set(lst[i]) & set(lst[k])]
mono = set(range(len(lst))) - set(n for ov in ovlp for n in ov)
ans = [lst[i] for i in mono] + [sorted(set(lst[i]+lst[k])) for i,k in ovlp]
print(ans)  # [['x', 'y'], ['a', 'b', 'd'], ['a', 'b', 'f']]

あるいは次のコードかな。
ppaul さん ご指摘の不具合があったため別コードにしました。

list1 = ['a', 'b', 'c']
list2 = ['d', 'f', 'e']
list3 = ['x', 'y', 'z']
list4 = ['b', 'f', 'g']

def merge_list(lst: [[]]) -> [[]]:
    prelen = len(lst)
    ovlp = [(i,k) for i in range(len(lst)) for k in range(i+1, len(lst)) if set(lst[i]) & set(lst[k])]
    mono = set(range(len(lst))) - set(n for ov in ovlp for n in ov)
    ans = [lst[i] for i in mono] + [sorted(set(lst[i]+lst[k])) for i,k in ovlp]
    return ans if len(ans) == prelen else merge_list(ans)

lst = [ls[:2] for ls in (list1, list2, list3, list4)]
print(merge_list(lst))  # [['x', 'y'], ['a', 'b', 'd', 'f']]

不具合のあった旧コードベースで修正すると次のコードになります。

list1 = ['a', 'b', 'c']
list2 = ['d', 'f', 'e']
list3 = ['x', 'y', 'z']
list4 = ['b', 'f', 'g']

def merge_list(lst: [[]]) -> [[]]:
    prelen = len(lst)
    ans = []
    for ls in lst:
        for i, elm in enumerate(ans):
            if set(ls) & set(elm):
                ans[i] = sorted(set(ls + elm))
                break
        else:
            ans.append(sorted(ls))
    return ans if len(ans) == prelen else merge_list(ans)

lst = [ls[:2] for ls in (list1, list2, list3, list4)]
print(merge_list(lst))  # [['a', 'b', 'd', 'f'], ['x', 'y']]

以下は不具合のある旧コードです。

list1 = ['a', 'b', 'c']
list2 = ['a', 'd', 'e']
list3 = ['x', 'y', 'z']
list4 = ['b', 'f', 'g']

lst = [ls[:2] for ls in (list1, list2, list3, list4)]
ans = []
for ls in lst:
    for j, elm in enumerate(ans):
        if set(ls) & set(elm):
            ans[j] = sorted(set(ls + elm))
            break
    else:
        ans.append(sorted(ls))

print(ans)  # [['a', 'b', 'd', 'f'], ['x', 'y']]

Comments

[ppaul]

下のコードでデータを変えてみました。
>>> list1 = ['a', 'b', 'c']
>>> list2 = ['d', 'f', 'e']
>>> list3 = ['x', 'y', 'z']
>>> list4 = ['b', 'f', 'g']
>>> lst = [ls[:2] for ls in (list1, list2, list3, list4)]
>>> ans = []
>>> for ls in lst:
... for j, elm in enumerate(ans):
... if set(ls) & set(elm):
... ans[j] = sorted(set(ls + elm))
... break
... else:
... ans.append(sorted(ls))
...
>>> print(ans)
[['a', 'b', 'f'], ['d', 'f'], ['x', 'y']]

[lehshell]

失礼しました。考慮から漏れました。m(__)m

[ppaul]

失礼ではありません。こうやって、すばやく手を動かす人は好感度が高いです。