pythonの変数名をfor文で判断したい

Question

pythonの変数名をif分で判断したいときにどのようにすればよいか教えてください。

3つの変数A1,A2,A3があるときに、for文内で今のループが0のときA1, 1のときA2, 2のときA3として判断したいです。

A1=10
A2=20
A3=30

B=0
for i in range(3):
    B += (i +1)* Ai # Aiはi=0のときA1, i=1のときA2,i=2のときA3を想定しています

print(B) # 140

Answer 1

一応変数名を動的に指定する事は出来ますが、普通はしません。特殊な用途向けですね。
普通はリストを使います。

A=[10,20,30]

B=0
for i in range(3):
    Ai = A[i]
    B += (i + 1) * Ai
print(B) # 140

さらに普通には、

A=[10,20,30]

B=0
for i, Ai in enumerate(A,1):
    B += i * Ai
print(B) # 140

内包表記を使うとB = sum(i*Ai for i, Ai in enumerate(A,1))

Comments

[redcatML]

回答ありがとうございます。
質問を簡単にするために、A1,A2,A3は定数で質問しましたが実は、
それらは配列を想定しています。
質問内容が変わってしまうので再度質問しなおさせていただきます。

Answer 2

A1=10
A2=20
A3=30

B=0
for i in range(3):
    Ai = locals().get(f'A{i+1}', 0)
    B += (i + 1) * Ai

print(B) # 140

Comments

[redcatML]

早速の回答ありがとうございます。
そのようにすればできるのですね。
ありがとうございました。