リスト変数をforループ内で更新 / 元に戻したい

Question

#問題


AtCoder ABC190 C問題より

#方針（途中まで、関係あるところのみ）
入力例1に従うと、

・C,DをCD = [list(map(int, input().split())) for _ in range(K)]として[[1, 2], [1, 3], [2, 3]]のような二重リストに格納する。

・itertools.productを用いて、CDの要素リストを、ビット全探索を用いてそのそれぞれの0番目または1番目の値で書き換えたものに更新し、printする。

・1つ出力したら書き換えられたCDを元に戻して、同様の作業を2^3回繰り返す。

#コード
ここではとりあえず、ビット全探索を用いてCDが2^3回期待通り書き換えられているかを、それぞれを全て出力することで確認することを目標とする：

from itertools import product

N, M = map(int, input().split()) #4 4

#[[1, 2], [1, 3], [2, 4], [3, 4]]
AB = [list(map(int, input().split())) for _ in range(M)]

K = int(input()) #3

#[[1, 2], [1, 3], [2, 3]]
CD = [list(map(int, input().split())) for _ in range(K)]

_CD = CD

#[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
for bit in list(product([0, 1], repeat = K)):
	CD = _CD
	for ind, to_place in enumerate(bit):
		CD[ind] = CD[ind][to_place]
	print(CD)

#出力エラー

[1, 1, 2]
Traceback (most recent call last):
    CD[ind] = CD[ind][to_place]
TypeError: 'int' object is not subscriptable

#試行錯誤
出力エラーによると、(0, 0, 0)のパターン([1, 1, 2])だけうまくいっていることがわかります。しかしながら、(0, 0, 1）のパターン([1, 1, 3])以降で失敗しています。どうやら一重リストに更新されてしまったCDを元input()した二重リストにうまく戻せていないようで、_CD = CDとして保存した_CDを用いて1つ目及び2つ目のforループの中や外に色々配置してみたのですが、どうやってもnot subscriotableエラーから抜け出せません。どのような箇所を修正すればうまくいくのでしょうか。素人質問で恐縮ですが、ご教授お願い申し上げます。

Answer 1

_CD = CD
CD = _CD
は_CDとCDが同じオブジェクトを見るようにしているだけです。

バックアップを取りたいなら、

_CD = CD.copy()
CD = _CD.copy()

としなければなりません。

Answer 2

まとめとしてですが、_CD = CD.copy()としてまずバックアップを取って、かつCD = _CDとしてしまうと結局二つのリストは同じオブジェクトを参照してしまうとのことで、forを用いてCDの各要素を_CDの各要素で書き換えるということをすることで実装することが出来ました：

from itertools import product

N, M = map(int, input().split()) #4 4

#[[1, 2], [1, 3], [2, 4], [3, 4]]
AB = [list(map(int, input().split())) for _ in range(M)]

K = int(input()) #3

# [[1, 2], [1, 3], [2, 3]]
CD = [list(map(int, input().split())) for _ in range(K)]

_CD = CD.copy()

for bit in list(product([0, 1], repeat = K)):
    for ind, to_place in enumerate(bit):
        CD[ind] = CD[ind][to_place]
    print(CD)
    for i in range(K):
    	CD[i] = _CD[i]

出力：

[1, 1, 2]
[1, 1, 3]
[1, 3, 2]
[1, 3, 3]
[2, 1, 2]
[2, 1, 3]
[2, 3, 2]
[2, 3, 3]