Python matplotlib /用线连接图

时间:2018-12-04 03:45:27

标签: python matplotlib

我有一个工具可以打开Twitter API的流,并根据推文的情绪分析推文。逻辑存储在一个类中。我正在使用全局变量来存储由类生成的值(所有整数)。随着推文的处理,它们会不断更新。

from tweepy.streaming import StreamListener
from textblob import TextBlob
from tweepy import Stream
from tweepy import OAuthHandler
import matplotlib.pyplot as plt
import re, json, time, datetime
import numpy as np
import tweepy

count = 0 
class listener(StreamListener):

    def on_data(self, data):
    ....(logic for processing data)....

    global count
    count = count + 1

我正在使用matplotlib每X秒绘制一次汇总值。为此,我是:

  1. 在子类外部使用while循环调用全局变量
  2. 将全局值绘制到图表上
  3. 将全局变量重置为其原始值(0)
  4. 每X秒重复一次

我的解决方案,用于每隔X秒将统计数据绘制到图表上

plt.ion() # for an interactive chart
read_frequency = 5 # how frequent we plot values to chart (seconds)

while True:

    time.sleep(read_frequency)
    now = datetime.datetime.now().isoformat()

    # prints aggregated data to terminal
    print(f'{now} Total: {count}')

    x = now
    y = count

    plt.plot(x, y, color = 'black', marker = 'o')

    plt.show()
    plt.pause(0.0001)

    # Reset aggregate data in global variables
    count = 0 

我现在遇到的问题是,由于我重新设置了全局变量(即ydata),因此绘图之间没有任何关系。因此,我得到的不是线型图,而是带有单个图的图。

My current chart

我如何为了在绘图之间划一条线而强制绘图之间的关系?

1 个答案:

答案 0 :(得分:0)

如果不保留汇总数据,则需要从图形本身获取数据。因此,您可以使用

获得最后绘制的(x,y)线
line = plt.gca().get_lines()[-1]
xd = line.get_xdata()
yd = line.get_ydata()

然后使用xdyd中的最后一个值以及您的新值绘制一条新线

x0, y0 = xd[-1], yd[-1]
plt.plot([x0, x], [y0, y], color = 'black', marker = 'o')