在for循环中调用对象

Question

我在努力简化我的代码。代码是用来在图形上画一些对角线的。以下是原始代码：

x = np.linspace(-5,6,50)
y1 = x
y2 = x+1
y3 = x+2
y4 = x+3
y5 = x+4
y6 = x-1
y7 = x-2
y8 = x-3
y9 = x-4

plt.plot(x,y1, linewidth=3, color='k')

plt.plot(x,y2, color='b')
plt.plot(x,y3, color='b')
plt.plot(x,y4, color='b')
plt.plot(x,y5, color='b')
plt.plot(x,y6, color='b')
plt.plot(x,y7, color='b')
plt.plot(x,y8, color='b')
plt.plot(x,y9, color='b')

我试图用以下“for”循环来简化最后几行：

for i in np.arange(2,10):
    plt.plot(x,yi, color='b')

Python无法识别i。正确的方法是什么？也许是lambda？我真的不确定

Answer 1

使用一个简单的for循环：

plt.plot(x, x, lw=3, color='k')

for i in range(-4, 5):
    if i != 0:
        plt.plot(x, x + i, color='b')