Matplotlib 绘图生成器

Question

当我这样做时它会绘图（在 Python 3 中）：

import matplotlib.pyplot as plt

a = [1,2,3,4,5]

plt.plot(range(5), a)
plt.show()

据我所知，在 Python 3 中，range() 返回一个迭代器，因此我上面的代码几乎是在迭代器上绘制的。现在我的问题是，为什么我不能这样做：

def myGen(iterable):
    for i in iterable:
        yield i
a = myGen([1,2,3,4,5])
plt.plot(range(5),a)
plt.show()

当我尝试这个时，我得到ValueError: x and y must have same first dimension

当您考虑它时，想要绘制两个生成器是有道理的。只是一个接一个的情节。这在matplotlib 中可行吗？

我知道there is a similar thread，很遗憾没有得到答复。

Answer 1

问题是，尽管 Python 3 中的 range 不是一个真正的列表，但它仍然提供了比您的自定义生成器更多的功能。特别是，它提供了__len__。这是至关重要的，因为 matplotlib 在后台将所有内容都转换为 numpy 数组，而 numpy 数组需要预先知道它们的大小。 Python 3 范围对象也提供了项目访问，所以它们真的很像普通的序列。您可以在the documentation 中看到它们不是简单的生成器。

我不确定你需要做些什么才能让你自己的自定义类在这里工作，但这可能比它的价值更麻烦。 Matplotlib 需要能够将您的数据转换为 numpy 数组，而 numpy 反过来需要比简单的“这个对象是可迭代的”更多的信息。它需要能够知道有多少数据并一次获取所有数据。