这给出了精确的期望图:import matplotlib.pyplot as plt
from collections import OrderedDict
T_OLD = {'10' : 'need1', '11':'need2', '12':'need1', '13':'need2','14':'need1'}
T_SRT = OrderedDict(sorted(T_OLD.items(), key=lambda t: t[0]))
plt.plot(map(int, T_SRT.keys()), map(lambda x: int(x[-1]), T_SRT.values()),'r')
plt.ylim([0.9,2.1])
ax = plt.gca()
ax.set_yticks([1,2])
ax.set_yticklabels(['need1', 'need2'])
plt.title('T_OLD')
plt.xlabel('time')
plt.ylabel('need')
plt.show()
对于Python 3.X,绘图行需要显式地将map()输出转换为列表:
^{pr2}$
在Python中,{X在Python中作为iterate}返回到一个列表中。在
绘图使用转换为int的字典键和need1或need2的最后一个元素,也转换为int。这取决于数据的特定结构,如果need1和{}的值需要更多的操作。在
绘制并更改轴限制后,程序只需修改y位置1和2处的记号标签。然后,它还会添加标题以及x和y轴标签。在
重要的一点是字典/输入数据必须进行排序。一种方法是使用OrderedDict。这里的T_SRT是一个按T_OLD中的键排序的OrderedDict对象。在
输出为:
对于T_OLD中的更多值/标签,这是一个更一般的情况。它假设标签总是'needX',其中X是任意数字。对于数字前面的任何字符串的一般情况,这很容易做到,尽管这需要更多的处理import matplotlib.pyplot as plt
from collections import OrderedDict
import re
T_OLD = {'10' : 'need1', '11':'need8', '12':'need11', '13':'need1','14':'need3'}
T_SRT = OrderedDict(sorted(T_OLD.items(), key=lambda t: t[0]))
x_val = list(map(int, T_SRT.keys()))
y_val = list(map(lambda x: int(re.findall(r'\d+', x)[-1]), T_SRT.values()))
plt.plot(x_val, y_val,'r')
plt.ylim([0.9*min(y_val),1.1*max(y_val)])
ax = plt.gca()
y_axis = list(set(y_val))
ax.set_yticks(y_axis)
ax.set_yticklabels(['need' + str(i) for i in y_axis])
plt.title('T_OLD')
plt.xlabel('time')
plt.ylabel('need')
plt.show()
此解决方案使用re.findall查找标签末尾的数字,以适应多位数的可能性。前面的解决方案只取字符串的最后一个部分,因为数字是个位数。它仍然假定用于绘制位置的数字是字符串中的最后一个数字,因此[-1]。同样,对于python3.X,映射输出显式地转换为list,在python2.7中不需要步骤。在
现在,通过使用set选择唯一的y值,然后通过将字符串'need'与其对应的整数串联来重命名它们的标签。在
y轴限值设为最小值的0.9和最大值的1.1。其余的格式和以前一样。在
此测试用例的结果是: