我正在尝试编写一个示例用户定义函数，该函数将进行绘图，但存在问题

Question

import numpy as np
import matplotlib.pyplot as plt
x=np.array([1,2,3,4,5,6])
def linear(a,b):
    return a*x+b
    plt.plot(x,linear(a,b))
    plt.show()

linear(2,4)

它只是给我输出[6,8,10,12,14,16]但不是一个情节。我看不出有什么不对。

Answer 1

您在使用之前使用return。将您的代码更改为以下内容：

import numpy as np
import matplotlib.pyplot as plt
x=np.array([1,2,3,4,5,6])
def linear(a,b):
    return a*x+b   
plt.plot(x,linear(2,4))
plt.show()