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

时间:2018-11-04 16:25:33

标签: python matplotlib

我想通过按键盘上的任意键从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()

1 个答案:

答案 0 :(得分:0)

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

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

当心:单击鼠标也会关闭窗口。

围绕它的非优雅方式:

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