python - 绘制非线性方程的解对

Question

我尝试绘制对 (x,y)，它们是方程 y^2 + y = x^3 + y(x^2) 的解，使用以下代码:

import numpy as np
y = np.arange(0,12,.01)
x = np.arange(0,18,.01)
for i in enumerate(y):
    for j in enumerate(x):
        if (i**2)+i == j**3 + i*(j**2):
            plot(i,j)

但是，我收到以下错误:“TypeError:** 或 pow() 不支持的操作数类型:'tuple' 和 'int'”

任何人都可以帮助我弄清楚如何绘制解决方案？

我还有另一个问题:是否可以将每对解决方案存储在一个变量中，以便我们将来可以使用特定的解决方案对？

Answer 1

enumerate 产生 (index, value) 元组，您可以使用解包在循环头中分别获取每个元组:

for i_ind, i in enumerate(y):
    for j_ind, j in enumerate(x):