在不使用 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)
 

这正是我想要的。

但是，我可以通过某种方式做到这一点 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)