将列表绘制为图形错误

Question

我试图绘制一个图表，y轴是两个列表的函数，但是我得到一个错误。请注意，time，firstlist和secondlist都具有相同的长度

plt.plot(time, (firstlist[i]**2 + secondlist[i]**2) for i in range(len(firstlist)))

我得到的错误是：

SyntaxError: Generator expression must be parenthesized if not sole argument

我在哪里错过了括号？

Answer 1

试试这个

plt.plot(time, [(firstlist[i]**2 + secondlist[i]**2) for i in range(len(firstlist[1:]))])