绘制基本矢量化函数

Question

我试图让我的函数在 python 列表间隔上绘制一个矢量化函数，并为该图设置一个 name/title。我对 numpy 和 matplotlib 等 python 包相当陌生。我希望有人能对我的代码有所了解：

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline    

def test_plot(x,interval,title):
    x = np.vectorize(x)
    interval = [0,interval+1]
    ttl = plt.title(title)
    plt.plot(x,interval)
    return plt.plot(x,interval)    

test_plot([-1,0,1,2,3,4],2,'test')

我收到此错误：

ValueError: x and y must have same first dimension

注意：我需要 interval 成为 list。

Answer 1

我不太明白你要绘制什么。

查看文档后，可以找到：

numpy.vectorize 需要一个 pyfunc 初始化。

例如：

>>> def myfunc(a, b):
...     "Return a-b if a>b, otherwise return a+b"
...     if a > b:
...         return a - b
...     else:
...         return a + b

然后启动你的矢量化函数

>>>vfunc = np.vectorize(myfunc)
>>>vfunc([1, 2, 3, 4], 2)
array([3, 4, 1, 2])

参数 pyfunc 必须是 callable 。传递一个列表似乎毫无意义。

至于matplotlib.pyplot.plot。

此函数要求 x 和 y 具有相同的维度才能绘制。

但是你的通行证是一个函数和两个元素列表导致 ValueError

Answer 2

像 KIDJournety 一样，我也不太了解您想要什么，但如果我正确理解您的 cmets，这可能会起作用：

def test_plot(x, interval, title): import matplotlib.pyplot as plt plt.title(title) plt.plot(x, interval) plt.show()

test_plot([1,2,3], [7,6,5], "hello, world!") 运行良好。 如果这不能满足您的需求（矢量化），请在下面发表评论，我会尽力相应地调整代码。

现在由于一些实际原因导致您的原始代码无法正常工作： 您传递了一个包含 6 个元素（-1 到 4）的列表或向量作为图形的 x 值，但只有 2 个元素（0 和区间+1，在您的示例中为 3）作为您的 y 值。您糟糕的代码只能配对 (-1, 0) 和 (0,3)。其余的 x 坐标没有要配对的对应 y 坐标，因此代码失败。