用户输入的Python交互式绘图

我在使用输入时与绘图交互时遇到问题

import matplotlib.pyplot as plt
x = [1, 2]
y = x
plt.plot(x, y)
plt.show()
a = input('Enter:')

在这里，在执行plt.show（）之后，会弹出一个绘图窗口，但在我向“a”提供一些用户输入之前，该窗口一直没有响应。 我想看一下绘图，然后我想把输入传递给“a”

请在打印前添加这条线，它将启用您的交互模式

请参阅文档：

def ion():
    'Turn interactive mode on.'
    matplotlib.interactive(True)
    install_repl_displayhook()

def ion():
    'Turn interactive mode on.'
    matplotlib.interactive(True)
    install_repl_displayhook()