rangeとint型の処理について

Question

matplotlibで三本ラインが書かれたグラフを描画したいのですが、xに１～５の数をrangeで与え、yはxの2乗の数を付与したいです。
グラフの2本目3本目のグラフはそれぞれxの1/2,1/3と簡易的なグラフを描きたいのですがrangeをｙに代入することはできないのでしょうか？

import matplotlib.pyplot as plt
%matplotlib inline

x = range(1,6)
y = x **2  

plt.plot(x, y)
plt.plot(x, y/2)
plt.plot(x, y/3)

plt.show()

エラー

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-18-d99ce3c9b342> in <module>
      3 
      4 x = range(1,6)
----> 5 y = x **2
      6 
      7 plt.plot(x, y)

TypeError: unsupported operand type(s) for ** or pow(): 'range' and 'int'

Answer 1

rangeオブジェクトを直接二乗することはできないので、内包表記で対応するなど必要です。
また、その後のy/2なども配列の各要素に対して四則演算をかけるには、内包表記・map・numpyなど一手間が必要です。
とりあえず、numpyのアレイ化したものがこちら。

import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

x = np.array(range(1,6))
y = x**2

plt.plot(x, y)
plt.plot(x, y/2)
plt.plot(x, y/3)

plt.show()

numpy使いたくない場合はこちら

import matplotlib.pyplot as plt
%matplotlib inline

x = [*range(1,6)]
y = [X**2 for X in x]
y2 = [(X**2)/2 for X in x]
y3 = [(X**2)/3 for X in x]
plt.plot(x, y)
plt.plot(x, y2)
plt.plot(x, y3)

plt.show()

Answer 2

rangeそのままではできないので、以下のようにnumpy.arrayにしてやればよいです。

import numpy as np
:
x = np.array(range(1,6))