python、リスト、for文

Question

pythonで
{3, 7, 0, 8, 4, 1, 9, 6, 5, 2}で初期化される大きさ10の整数型配列を宣言し、最初は参照する要素番号を0とし、この参照する要素番号の配列要素の値を表示し、次にその配列要素の値を次の参照する要素番号とし、この次の参照する要素番号の配列要素の値を表示し、さらにその配列要素の値を次の参照する要素番号とし、……を10回繰り返すプログラムを作成したいです。

ソースコードは、
a=[3, 7, 0, 8, 4, 1, 9, 6, 5, 2]
for i in range(10):
if(i==0):
print(a[0])
else:
a[i]=a[a[i-1]]
print(a[i])

です。どこのコードをいじればいいかが分かりません。

実行結果は
3
8
5
1
8
5
5
5
5
5

になりますが、以下のようにしたいです。

3
8
5
1
7
6
9
2
0
3

Comments

[Orlofsky]

質問は修正できます。
ここの掲示板ではコードや処理結果は Markdownの[コード]を使ってください。
https://teratail.com/help/question-tips#questionTips37

Answer 1

a = [3, 7, 0, 8, 4, 1, 9, 6, 5, 2]

idx = 0
for _ in range(len(a)):
    print(a[idx])
    idx = a[idx]

#
3
8
5
1
7
6
9
2
0
3

※ Python 3.8 で導入された :=(walrus operator) を使う場合。

idx = 0
for _ in range(len(a)):
    print(idx := a[idx])