Python多线程中多次调用input()引发错误的解决求助

问题根源 #

MacOS的Cocoa框架强制要求所有UI相关操作（包括Matplotlib依赖的事件循环）必须在主线程执行。你用子线程循环调用input()时，后续输入会间接触发Matplotlib的UI事件处理逻辑，导致Cocoa在非主线程调用nextEventMatchingMask，违反线程规则抛出异常。单次input()能正常运行，是因为子线程在主线程动画启动前就完成了输入，未触发后续冲突。

可行解决方法 #

方法1：使用Matplotlib内置交互组件替代子线程输入

直接在主线程的Matplotlib界面中添加输入控件（比如文本框），所有操作都在主线程事件循环内处理，从根源避免线程冲突：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.widgets import TextBox

fig, ax = plt.subplots()
plt.subplots_adjust(bottom=0.2)

radius = 1.0
theta = 0.0
point, = ax.plot(radius*np.cos(theta), radius*np.sin(theta), 'ro')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)

def update_radius(text):
    global radius
    try:
        radius = float(text)
    except ValueError:
        # 输入无效时保持原半径
        pass

# 添加下方输入框
axbox = plt.axes([0.1, 0.05, 0.8, 0.075])
text_box = TextBox(axbox, 'Radius:')
text_box.on_submit(update_radius)

def animate(frame):
    global theta
    theta += 0.05
    point.set_data(radius*np.cos(theta), radius*np.sin(theta))
    return point,

ani = plt.FuncAnimation(fig, animate, interval=50, blit=True)
plt.show()

方法2：主线程内非阻塞检查输入

利用select模块在动画更新函数中监听标准输入，所有逻辑都在主线程执行，无需子线程：

import matplotlib.pyplot as plt
import numpy as np
import sys
import select

fig, ax = plt.subplots()
radius = 1.0
theta = 0.0
point, = ax.plot(radius*np.cos(theta), radius*np.sin(theta), 'ro')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)

def animate(frame):
    global theta, radius
    # 非阻塞检查是否有输入
    if select.select([sys.stdin], [], [], 0)[0]:
        line = sys.stdin.readline().strip()
        try:
            radius = float(line)
        except ValueError:
            pass
    theta += 0.05
    point.set_data(radius*np.cos(theta), radius*np.sin(theta))
    return point,

ani = plt.FuncAnimation(fig, animate, interval=50, blit=True)
plt.show()

方法3：子线程+队列传递输入（保留终端输入方式）

如果必须保留终端输入，让子线程仅负责读取输入并放入队列，主线程在动画更新时从队列取数据，避免子线程触发UI事件：

import matplotlib.pyplot as plt
import numpy as np
import threading
import queue

radius_queue = queue.Queue()

def input_thread():
    while True:
        try:
            line = input("Enter radius: ")
            radius_queue.put(line)
        except EOFError:
            break

# 启动后台子线程
threading.Thread(target=input_thread, daemon=True).start()

fig, ax = plt.subplots()
radius = 1.0
theta = 0.0
point, = ax.plot(radius*np.cos(theta), radius*np.sin(theta), 'ro')
ax.set_xlim(-5, 5)
ax.set_ylim(-5, 5)

def animate(frame):
    global theta, radius
    # 处理队列中的所有新输入
    while not radius_queue.empty():
        line = radius_queue.get()
        try:
            radius = float(line)
        except ValueError:
            pass
    theta += 0.05
    point.set_data(radius*np.cos(theta), radius*np.sin(theta))
    return point,

ani = plt.FuncAnimation(fig, animate, interval=50, blit=True)
plt.show()

内容的提问来源于stack exchange，提问作者Lawrence