首页 > 解决方案 > 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,所以图之间没有关系。因此,我得到的不是线型图表,而是带有单独图的图表。

我现在的图表

为了在它们之间画一条线,我如何强制地块之间建立关系?

标签: pythonmatplotlib

解决方案


如果不保留聚合数据,则需要从图形本身获取数据。所以你可以得到最后一条 (x,y) 绘制的线

line = plt.gca().get_lines()[-1]
xd = line.get_xdata()
yd = line.get_ydata()

xd然后使用 and 中的最后一个值和yd您的新值绘制一条新线

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

推荐阅读