numpyを使用して配列に定数を乗算する繰り返しを行いたい

Question

配列1,2,3に定数3を乗算して1,2,3,3,6,9といったように10回繰り返して数列を表示させたいのですが、変数が更新されなかったり、綺麗な数列が作れません。

コード(インタプリンタで実行)
import numpy as np
a = np.array([1,2,3])
iters_num = 10
for a in range(iters_num):
    a * 3
    print(a)

出力
0
0
3
1
6
2
9
3
...

となり、綺麗に1,2,3,3,6,9というような数列を表示できません。
どうしたらいいしょうか？教えてください。

Answer 1

そのプログラムでは、出力は
0
1
2
3
4
5
6
7
8
9
になるような気がするのですが・・・・それは置いておいて、

・a=... の下の行に print(a)を入れる
・for の 後の a を 違う変数名(例えば b ) に変更する
・a3 を a=3 にする。
という対応が必要かと。全部入れるとと↓のような感じ。

import numpy as np
print(a)
a = np.array([1,2,3])
iters_num = 10
for b in range(iters_num):
    a *= 3
    print(a)

出力は、下記のようになります。
[1 2 3]
[3 6 9]
[ 9 18 27]
[27 54 81]
[ 81 162 243]
[243 486 729]
[ 729 1458 2187]
[2187 4374 6561]
[ 6561 13122 19683]
[19683 39366 59049]
[ 59049 118098 177147]

Comments

[LouiS0616]

インタプリタでポチポチ実行しているようなので、計算式の結果も出力されているようです。

[saAsectron36]

ありがとうございます。大変助かりましたm(__)m

Answer 2

ちょっと、知識が足りなすぎます。

import numpy as np
a = np.array([1,2,3])       # A
iters_num = 10
for a in range(iters_num):  # B
    print(a * 3)
    print(a)

コメントのAの箇所とBの個所で、それぞれaは別物です。

---

Aの個所のa

a = np.array([1,2,3])

おそらくあなたが予期しているとおり、numpyのリストがaに代入されています。

Bの個所のa

for a in range(iters_num):  # B

range(iters_num)に呼応して、ループごとに0,1,2... になる整数です。

---

さらにもう一か所。

for a in range(iters_num):  # B
    a * 3   # ←ここ
    print(a)

←ここ　と書いた部分について。
3をかける計算は確かに実行されていますが、その結果が即座に捨てられています。
計算結果を代入しないと意味がないです。a = a * 3こんな感じ。

---

単純にリストの各値を三倍したいだけなら、以下のようにすればよいです。

import numpy as np
a = np.array([1, 2, 3])
a = a * 3
for elem in a:
    print(elem)

"""出力
3
6
9
"""

---

1,2,3,3,6,9と10回繰り返して表示させたい

目的が大いに謎ですが、それならこう。

import numpy as np
a1 = np.array([1, 2, 3])
a2 = a1 * 3
iter_num = 10
for _ in range(iter_num):
    for elem in a1:
        print(elem, end=', ')
    for elem in a2:
        print(elem, end=', ')

print('')

"""出力
1,2,3,3,6,9,1,2,3,3,6,9,　...以下略
"""