如何使用numpy/python进行“连锁反应”？

import numpy as np
import matplotlib.pyplot as plt

# 定义反应方程
def reaction(y, t):
    A, B, C = y  # 反应物浓度
    k1 = 0.1  # 反应速率常数1
    k2 = 0.2  # 反应速率常数2

    # 定义微分方程
    dA_dt = -k1 * A
    dB_dt = k1 * A - k2 * B
    dC_dt = k2 * B

    return [dA_dt, dB_dt, dC_dt]

# 定义初始条件
y0 = [1.0, 0.0, 0.0]  # A的初始浓度为1，B和C的初始浓度为0

# 定义时间步长和模拟时间
dt = 0.01
t = np.arange(0.0, 10.0, dt)

# 使用数值方法求解微分方程组
y = np.zeros((len(t), 3))
y[0] = y0
for i in range(1, len(t)):
    y[i] = y[i-1] + dt * np.array(reaction(y[i-1], t[i-1]))

# 绘制浓度随时间的变化曲线
plt.plot(t, y[:, 0], label='A')
plt.plot(t, y[:, 1], label='B')
plt.plot(t, y[:, 2], label='C')
plt.xlabel('Time')
plt.ylabel('Concentration')
plt.legend()
plt.show()