Ошибка гипотезы Коллатца Matplotlib
Я надеюсь, что кто-то может помочь с этим проектом для школы. Я пытаюсь построить (используя matplotlib на rep.it) итерации в диапазоне 1000. Я уверен, что есть лучший способ сделать это, и я в растерянности. Я чувствую, что я так близко, но я получаю эту ошибку: локальная переменная 'current_number' упоминается до назначения.
import math
import matplotlib as mpl
import matplotlib.pyplot as plt
def collatz(x):
x_coordinates = []
y_coordinates = []
count = 1
curent_number = 0
while x != 1:
if x % 2 == 0:
x=x/2
count=count+1
current_number=current_number+1
x_coordinates.append(x)
y_coordinates.append(current_number+1)
else:
x=(x*3)+1
count=count+1
current_number=current_number+1
x_coordinates.append(x)
y_coordinates.append(current_number)
return count
return current_number
return y_coordinates
return x_coordinates
collatz(1000)
plt.plot(x_coordinates, y_coordinates)
plt.savefig('plot.png')