python - matplotlib - 运行时在事先不知道的情况下添加多行

Question

我试图在一个图形上添加多行，但事先不知道行数。我目前有一个类，其中的 x 和 y 值用于单个 session 中的行。

我不确定如何在同一图中为每个新 session 添加新行。创建具体的关联。

在我的主函数中，我有以下代码。

fig = plt.figure()
ax = fig.add_subplot(111)
line,= plt.figure().plot(0,0)

在我的类(class)中。我有以下代码。

class Session:
x = []
y = []
# I think I should add a line here... but I am not sure 
# how to make this association to the main.

对于每个 session ，它存储 x 和 y 值，我可以通过方法检索这些值。这部分很简单，但将每条线关联到同一个图表是我遇到的麻烦。我该如何解决这个问题？

Answer 1

您可以多次调用.plot()。我添加了如何更改线条颜色的示例。我将把样式留给你。

import matplotlib.pyplot as plt

fig = plt.Figure()
ax = fig.add_subplot(111)

colors = ('b','g','r','c','m','y','k','w',)
sessions = (sess1, sess2, sess3)
for sess, color in zip(sessions, colors):
    ax.plot(sess.x, sess.y, color=color)

<小时/>

如果您想为所有线条使用和/或重复使用一组特定的颜色 itertools.cycle很短的工作:

import itertools as it

colors = ('b','g','r',)
sessions = (sess1, sess2, sess3, sess4, sess5, sess6)
for sess, color in zip(sessions, it.cycle(colors)):
    ax.plot(sess.x, sess.y, color=color)