不使用 for 循环绘制多个图形

Question

因此，我的问题可能与标题中的内容不完全相同。

我有一个功能

y = a*x + b

我想用不同的 b 值绘制 y。

我知道我可以做到以下几点：

import numpy as np
import matplotlib.pyplot as plt

a = 2
x = np.array([0,1,2,3,4])
b = 0

for i in range(10):
    y = a*x + b
    b = b+1
    plt.plot(x,y)

而且 returns 正是我想要的。

但是，我可以通过使用 b = np.array([0,1,2,3,4,5,6,7,8,9])？那么，我的代码可能类似于：

import numpy as np
import matplotlib.pyplot as plt

a = 2
x = np.array([0,1,2,3,4])
b = np.array([0,1,2,3,4,5,6,7,8,9])

y = a*x + b
plt.plot(x,y)

Answer 1

是的，您可以使用矩阵运算用运算结果创建二维矩阵 y = a*x + b。

a = 2
x = np.array([0,1,2,3,4])
b = np.array([0,1,2,3,4,5,6,7,8,9])

y = a*x[:,None]+b
plt.plot(x, y)

编辑： 我正在展示 @Quang Hoang 提供的解决方案，它比我的简单得多。原始代码是：

y = np.tile(a*x, (b.size,1)) + b[:,np.newaxis]
plt.plot(x, y.T)

不使用 for 循环绘制多个图形

Plot multiple graphs without using a for loop

numpy

graph

matplotlib

algebra