在不使用for循环的情况下绘制多个图

问题描述

所以，我的问题可能不完全是标题中的内容。

我有一个功能

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,4,5,6,7,8,9])？因此，我的代码可能类似于：

import numpy as np
import matplotlib.pyplot as plt

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

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

解决方法

是的，您可以使用矩阵运算来创建2D矩阵，其结果为y = a*x + b。

a = 2
x = np.array([0,1,2,3,4])
b = np.array([0,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)