从数据文件更新Python绘图图例

时间:2017-07-31 11:15:18

标签: python matplotlib

我有几个数据文件,每个文件有三列:

    k t x
    1 0 0
    1 1 10
    1 2 20

等,用于不同的k值。

我想在 Python 中制作x vs t的图,这非常简单。现在,我想制作一个传奇,显示" k = 1"。现在,我能想到的最简单的方法是手工编写每个图例(非常无聊!)。

是否有任何有效的方法可以自动更新图例,即传说" k =?"在哪里`?'将从数据文件中自动获取值?

我尝试的例子是基本的绘图,所以我没有在这里再现,而对于传说部分,我真的不知道如何继续!

请帮助......

1 个答案:

答案 0 :(得分:1)

请参阅以下最小示例。由于k似乎与整个文件相同,因此您只需使用k的第一个条目将其放入图例中。

import io
import matplotlib.pyplot as plt
import numpy as np

u = u"""k t x
        1 0 0
        1 1 10
        1 2 20"""

k, t, x = np.loadtxt(io.StringIO(u), skiprows=1, unpack=True)

plt.plot(t,x, label="k={}".format(k[0]))
plt.legend()
plt.show()