Несколько проблем с вашим кодом:
zm
должен быть массивом, но вместо этого это целое число, которое перезаписывается каждый цикл с возвращением i**2
, - Инструкция
plot()
должна находиться вне цикла, - Вам не нужен цикл for, вы можете сделать квадрат массива с помощью оператора
**
.
Я думаю, это то, что вы ищете:
import numpy as np
import matplotlib.pyplot as plt
xx = np.arange(1, 5)
zm = xx**2
plt.figure()
plt.plot(xx,zm,'r')
plt.show()
![enter image description here](https://i.stack.imgur.com/Yu7wb.png)
Кстати, я полагаю, вы имели в виду square
, а не square root
.
Надеюсь, это поможет.