二次元配列でその要素があればすでにある配列に数字を格納したい

Question

二次元配列でその要素があればすでにある配列に数字を格納したいです。

今

X = [[1, 2], [1, 3], [4, 5]]

という配列があります。
この配列を別の配列Yに入れ替えたいです。入れ替える条件指定として、
二次元配列で既出の数字が入っている配列があれば、その配列に数字を格納したいです。
配列Xに対しては、Yは

Y = [[1, 2, 3], [4, 5]]

という形にしたいです。

for i in range(3):
        if (X[i][1] in Y) or (X[i][2] in Y):
           Y.append(X[i][0:2])

と書いたのですが、うまくいかず...。
どう書けば良いのでしょうか？

Comments

[_Victorique__]

辞書型にすると重複を１つに出来るのでそれで試してみてください。

[退会済みユーザー]

それは何を辞書型にするのでしょうか？できればサンプルコードなどを提示してもらえますか？

Answer 1

集合を使うといいと思いますよ。

X = [[1, 2], [1, 3], [4, 5]]
temp = [ set( X[0] ) ]
for x in X:
    xSet = set(x)
    for y in temp:
        if not y.isdisjoint(xSet):
            y |= xSet
            break
    else:
        temp.append( xSet )
Y = [ list(x) for x in temp ]

Comments

[退会済みユーザー]

ありがとうございます。
もしXがX = [[7,1, 2], [7,1, 3], [8, 4, 5]]と３つずつの要素を持っていて、要素の１番目以外（この例では７,8以外）の[1, 2], [1, 3], [4, 5]　でまた同じことをするにはどうしたらいいのでしょうか？Lhankor_Mhyさんのコードを色々といじって見たのですが、うまくいかなくて...

[Lhankor_Mhy]

スライスすればいいのでは？

[退会済みユーザー]

はい、temp = [ set( X[0][1:3] ) ]　とやって見たのですが、スライスを使わなかったX = [[1, 2], [1, 3], [4, 5]]の時のYと結果が変わってしまい、同じ結果になりませんでした...

[Lhankor_Mhy]

全部の要素をスライスしないといけないでしょうね。

[退会済みユーザー]

X = [[0, 1, 2], [0, 1, 3], [1, 3, 2], [1, 3, 4], [0, 4, 5], [1, 3, 5], [0, 1, 4], [1, 3, 5]]
for i in range(7):
if X[i][0] == 0:
temp = [ set( X[0][1:3] ) ]
print(temp)
for x in X[i][1:3]:
print(X[i][1:3])
xSet = X[i][1:3]
for y in temp:
if not y.isdisjoint(xSet):
y |= xSet
break
else:
temp.append( xSet )
Y = [ list(x) for x in temp ]

[退会済みユーザー]

と書いたのですが、
TypeError: unsupported operand type(s) for |=: 'set' and 'list'　とエラーが出てしまい...。全部の要素をスライスとはこのような意味ではないのでしょうか？

[Lhankor_Mhy]

ループのネストで処理しない方がいいと思います。
先に全要素スライスしてから処理しては？

[退会済みユーザー]

うーん、先に全要素スライスとはどういう意味でしょうか？コードで示してもらってもいいですか？

[Lhankor_Mhy]

ループを書いて全要素をスライスした新しいリストを作るということですよ。
配列内包でもいいと思います。