Python的圖例從數據文件

Question

更新我有幾個數據文件每三列：

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

等，對於不同的k值。

我想製作一個x vs t的圖，Python，這很容易。現在，我想創建一個顯示「k = 1」的圖例。現在，我能想到的最簡單的方法就是手寫每個圖例（很無聊！）。

是否有自動更新圖例，即，圖例「K =？」的任何有效的方式'？'在哪裏？會自動從數據文件中提取值？

我已經試過的例子是基本的繪圖，使我沒有在這裏重現，併爲傳奇的一部分我真的不知道該如何着手！

請幫助...

Answer 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()