Collatz猜想(3X+1)动画实现问题排查及功能优化咨询

问题1：空列表初始化报错的原因与解决方案 #

你直接对空列表使用y[i]赋值会触发索引越界，是因为空列表没有预分配对应下标的元素，列表仅支持通过append()方法在末尾追加新元素。
你可以直接用列表迭代生成Collatz序列，无需提前固定长度，也不需要处理np.empty带来的残留0值，修改后的collatz函数逻辑如下：

def collatz(y_start):
    y = [y_start]
    current = y_start
    while current != 1:
        if current % 2 == 0:
            current = even(current)
        else:
            current = odd(current)
        y.append(current)
    return np.array(y)

生成y序列后，x轴范围可以直接设置为plt.xlim(0, len(y)-1)，适配不同初始值的序列长度。

问题2：动画点上方显示数值的实现 #

可以通过matplotlib的annotate接口逐帧添加数值标注，每次更新帧时同步显示当前点的数值，注意设置纵向偏移避免标注和折线重合。

问题3：while语句死循环的原因与修正 #

旧版本死循环的核心问题是while循环内部没有修改迭代变量i的值，y[i]永远是进入while循环时的初始值，循环条件y[i] != 1永远成立，永远不会退出。
实际上生成Collatz序列不需要for和while嵌套，直接用while判断当前数值是否等于1即可，参考问题1中给出的collatz函数写法即可解决。

修复全部问题后的完整可运行代码 #

import numpy as np
from matplotlib.animation import FuncAnimation
from matplotlib import pyplot as plt

def odd(num):
    return (num*3)+1

def even(num):
    return num // 2

y_start = 33

def collatz(start):
    y = [start]
    current = start
    while current != 1:
        if current % 2 == 0:
            current = even(current)
        else:
            current = odd(current)
        y.append(current)
    return np.array(y)

y = collatz(y_start)
x = np.arange(len(y))

fig = plt.figure()
plt.xlim(0, len(y)-1)
plt.ylim(1, max(y)*1.1) # 给顶部标注预留空间

draw, = plt.plot([],[], marker='o', markersize=4)
annotations = [] # 存储标注对象，避免重复叠加

def update(idx):
    # 清除之前的标注
    for ann in annotations:
        ann.remove()
    annotations.clear()
    # 更新折线数据
    draw.set_data(x[:idx+1], y[:idx+1])
    # 添加当前点的数值标注
    ann = plt.annotate(f'{y[idx]}', 
                      xy=(x[idx], y[idx]),
                      xytext=(0, 5), # 向上偏移5个像素
                      textcoords='offset points',
                      ha='center', fontsize=8)
    annotations.append(ann)
    return draw, *annotations

a = FuncAnimation(fig, update, frames=len(y), interval=300, repeat=False)
plt.show()

内容的提问来源于stack exchange，提问作者Maggie