Pythonのfor文で配列に値を入れていきたい

Question

Pythonでx軸に自然数、y軸にその平方根を取りグラフ描画したく考えております。以下の様に、x軸の配列とy軸の配列を作成後、for文で値を一つずつ入れていこうとしますと、エラーが出ます。
対処につき、ご教示下さい。

from math import *
import matplotlib.pyplot as plt
import numpy as np

x_axis = np.linspace(2,20,num=19)
y_axis = x_axis
y_axis = np.array(y_axis)

for i in x_axis:
    print(i)
    y_axis[i] = sqrt(i)
    
plt.plot(x_axis,y_axis)

【エラー】
2.0

IndexError Traceback (most recent call last)
<ipython-input-27-5e6611feacbc> in <module>()
9 for i in x_axis:
10 print(i)
---> 11 y_axis[i] = sqrt(i)
12
13 plt.plot(x_axis,y_axis)

IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

Answer 1

iには0などのインデックス番号ではなく、2.0などの実際の値がセットされます。
以下のいずれかを行ってください。

# 略
# 方法1 enumerate でインデックスと値を走査
for i,v in enumerate(x_axis):
    print(i) # 0など
    y_axis[i] = sqrt(v)

# 方法2 np.sqrtを使って一括セット
y_axis = np.sqrt(x_axis)
# 略

Comments

[_hh]

方法１で試させて頂き、できました！有難うございます。