时间戳

首页 > 解决方案 > 通过键盘输入关闭 plt.show()

python - 通过键盘输入关闭 plt.show()

问题描述

我想通过按键盘上的任意键从 plt.show() 关闭窗口。目前，我不知道为什么，我需要按两次键才能关闭窗口。

这是我的代码：

# -*- coding: utf-8 -*-
import datetime
import os
import psycopg2
import matplotlib.pyplot as plt


try:
    conn = psycopg2.connect("dbname='mydb' user='mysur' host='localhost' password='mypw'")
except psycopg2.DatabaseError, ex:
    print 'I am unable to connect the database: ' + str(ex)

cur = conn.cursor()




cur.execute("select day, avg_price from day_sumary where day > '2018-05-20' and coin_nome = 'LTC'")
records = cur.fetchall()

coin_nome_sql = 'LTC'

cur.execute("select day, amount from day_sumary where day > '2018-05-20' and coin_nome = '"+ coin_nome_sql+"'")
records = cur.fetchall()
print(coin_nome_sql)
day, amount = zip(*records)
# graph code
plt.plot(day, amount, label= coin_nome_sql)
# draw a grid
plt.grid()

conn.commit()
cur.close()
conn.close()
# set title, X/Y labels
plt.title("amount per day")
plt.xlabel("day")
plt.ylabel("amount")
plt.legend()
plt.show()
plt.waitforbuttonpress(0)
plt.close()

标签： pythonmatplotlib

解决方案

您可以使用plt.draw()代替plt.show()，如

    plt.draw()
    plt.waitforbuttonpress(0)
    plt.close()

当心：它也会在鼠标点击时关闭窗口。

不优雅的方式：

    plt.draw()
    while True:
        if plt.waitforbuttonpress(0) == True:
            plt.close()
            break

推荐阅读