Matplotlib Python网格线切换按钮无法创建的问题排查

问题描述 #

在Matplotlib动画绘图中尝试添加点击切换网格线显示/隐藏的按钮，此前已成功实现动画轨道线条的切换功能，但网格线切换始终无效。怀疑问题与Matplotlib的轴设置、zorder或动画每帧重绘机制有关，尝试过修改透明度、颜色、调用ax.grid(False)、直接修改网格线可见性等多种方式均未解决。

尝试的代码版本1 #

ax_grid = fig.add_axes([0.85, 0.2, 0.1, 0.04])
gbutton = Button(ax_grid, 'Grid', color = '0.3', hovercolor='0.7')
# Define the grid visibility state
grid_visible = False  # Initialize the grid visibility state

def grid_lines(event):
    global grid_visible
    grid_visible = not grid_visible  # Toggle the state
    plt.sca(ax)  # Ensure we are modifying the main plot's axes
    ax.set_axisbelow(False) # trying to set it above
    fig.canvas.draw()
    if grid_visible:
        ax.grid(color='white')  # Show grid with properties
    else:
        ax.grid(color='black')  # Simply turn off the grid without extra arguments

    fig.canvas.draw_idle()  # Redraw the canvas

尝试的代码版本2 #

ax_grid = fig.add_axes([0.85, 0.2, 0.1, 0.04])
gbutton = Button(ax_grid, 'Grid', color = '0.3', hovercolor='0.7')
# Define the grid visibility state
grid_visible = False  # Initialize the grid visibility state

def grid_lines(event):
    global grid_visible
    grid_visible = not grid_visible  # Toggle state

    # Access grid lines directly and toggle their visibility
    for line in ax.get_xgridlines() + ax.get_ygridlines():
        line.set_visible(grid_visible)

    fig.canvas.draw_idle()  # Redraw the canvas

完整程序代码 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from matplotlib.widgets import Slider, Button
import scipy
from math import sqrt

plt.rcParams["figure.autolayout"] = True

print("Default text color is: ", plt.rcParams['text.color'])
plt.rcParams.update({'text.color': "white"})  # changing default text colour to white

dt = 0.1
numsteps = 10000
pi = scipy.constants.pi
G = 4.30091e-3  # AU^3 * M_sun^-1 * yr^-2
wA = 3.0
wB = 3.0

thetaA = 0
thetaB = thetaA + pi  # to put the other store on the opposite end of starA

# initialise variables
r = 5
mA = 10  # mass in solar mass
mB = 10
M = mA + mB

x_valA, y_valA = [], []
x_valB, y_valB = [], []

# Create the animation, fig represents the object/canvas and ax means it is the area being plotted on
fig, ax = plt.subplots(figsize=(8, 8), dpi=100)
ax.set_xlim(-10, 10)
ax.set_ylim(-10, 10)
ax.set_aspect('equal')
ax.set_facecolor("black")
fig.patch.set_facecolor("k")

# Create the stars and COM plot
starA, = ax.plot([], [], 'o', color='blue', markersize=10, label='Star A',zorder=9.5)
starB, = ax.plot([], [], 'o', color='red', markersize=10, label='Star B',zorder=9.5)
COM = ax.plot([0], [0], '+', color='white', markersize=5, label='COM',zorder=10)
orbitA, = ax.plot([], [], '-', color='cyan', alpha=0.5, label='Orbit A', zorder=5)
orbitB, = ax.plot([], [], '-', color='pink', alpha=0.5, label='Orbit B', zorder=5)
ax.grid(color='white', linestyle='--', linewidth=0.5, zorder=1)
leg = ax.legend(facecolor='k', labelcolor='w', fancybox=True, framealpha=0.6, loc='upper right', bbox_to_anchor=(1,1))
leg.set_zorder(20)

def orbit(r, mA, mB, M):
    global x_valA, y_valA, x_valB, y_valB, thetaA, thetaB, G, dt

    # Reset variables
    x_valA, y_valA = [], []
    x_valB, y_valB = [], []

    M = mA + mB

    rA = r * (mB / M)
    rB = r * (mA / M)

    # initial positions
    positionA = np.array([rA * np.cos(thetaA), rA * np.sin(thetaA)])  # Star A initial position
    positionB = np.array([rB * np.cos(thetaB), rB * np.sin(thetaB)])  # Star B initial position

    # SIMULATION LOOP
    for _ in range(numsteps):
        # Store positions for both stars
        x_valA.append(positionA[0])
        y_valA.append(positionA[1])
        x_valB.append(positionB[0])
        y_valB.append(positionB[1])

        # Update speed and angles for next positions
        wA = sqrt(G * M / rA * rA)
        wB = sqrt(G * M / rB * rB)
        thetaA += wA * dt  # update angle for starA
        thetaB += wB * dt  # update angle for starB

        # Calculate new positions based on updated angles
        positionA = np.array([rA * np.cos(thetaA), rA * np.sin(thetaA)])
        positionB = np.array([rB * np.cos(thetaB), rB * np.sin(thetaB)])

    # After simulation loop, update the orbit lines with the recorded paths
    orbitA.set_data(x_valA, y_valA)
    orbitB.set_data(x_valB, y_valB)

# initialising the data
def init():
    starA.set_data([], [])
    starB.set_data([], [])
    orbitA.set_data([], [])  # Clear the initial orbit paths
    orbitB.set_data([], [])
    
    return starA, starB, orbitA, orbitB

def update(frame):
    starA.set_data([x_valA[frame]], [y_valA[frame]])  # Pass as lists
    starB.set_data([x_valB[frame]], [y_valB[frame]])
    if orbit_lines_visible:
        orbitA.set_data(x_valA[:frame+1], y_valA[:frame+1])
        orbitB.set_data(x_valB[:frame+1], y_valB[:frame+1])
    return starA, starB, orbitA, orbitB

ani = FuncAnimation(fig, update, frames=numsteps, init_func=init, blit=False, interval=50)
plt.title("Binary Star System", fontsize=20, fontweight='bold')

def up(val):
    global r, mA, mB, M
    r = seperation_slider.val
    mA = mA_slider.val
    mB = mB_slider.val
    M = mA + mB
    orbit(r, mA, mB, M)  # updated values into function
    starA.set_markersize(mA_slider.val)
    starB.set_markersize(mB_slider.val)
    ani.event_source.stop()  # Stop the current animation
    ani.event_source.start()  # Restart the animation with updated orbit
    fig.canvas.draw_idle()
    
seperation_slider = Slider(ax=plt.axes([0.125, 0.02, 0.10, 0.04]), label='Seperation', valmin=1, valmax=15, valinit=r, valstep=1.11, facecolor='w')
mA_slider = Slider(ax=plt.axes([0.45, 0.02, 0.15, 0.04]), label="Mass A", valmin=0.1, valmax=100, valinit=mA, valstep=1.11, facecolor='b')
mB_slider = Slider(ax=plt.axes([0.80, 0.02, 0.15, 0.04]), label="Mass B", valmin=0.1, valmax=100, valinit=mB, valstep=1.11, facecolor='r')

seperation_slider.label.set_size(12)
mA_slider.label.set_size(12)
mB_slider.label.set_size(12)
mA_slider.vline.set_color('cyan')
mB_slider.vline.set_color('violet')
seperation_slider.vline.set_color('black')

seperation_slider.on_changed(up)
mA_slider.on_changed(up)
mB_slider.on_changed(up)

orbit(r, mA, mB, M)

ax_reset = fig.add_axes([0.85, 0.08, 0.1, 0.04])
rbutton = Button(ax_reset, 'Reset', color='0.3', hovercolor='0.7')

def reset(event):
    seperation_slider.reset()
    mA_slider.reset()
    mB_slider.reset()

rbutton.on_clicked(reset)

# Toggle for orbit lines visibility
orbit_lines_visible = False

def lines(event):
    global orbit_lines_visible
    if orbit_lines_visible:
        orbitA.set_alpha(0)  # Hide orbit A
        orbitB.set_alpha(0)  # Hide orbit B
    else:
        orbitA.set_alpha(0.5)  # Show orbit A
        orbitB.set_alpha(0.5)  # Show orbit B
    orbit_lines_visible = not orbit_lines_visible
    fig.canvas.draw_idle()
ax_lines = fig.add_axes([0.85, 0.14, 0.1, 0.04])
button = Button(ax_lines, 'Orbit Lines', color='0.3', hovercolor='0.7')
button.on_clicked(lines)

ax_grid = fig.add_axes([0.85, 0.2, 0.1, 0.04])
gbutton = Button(ax_grid, 'Grid', color = '0.3', hovercolor='0.7')
# Define the grid visibility state
grid_visible = False  # Initialize the grid visibility state

def grid_lines(event):
    global grid_visible
    grid_visible = not grid_visible
    
    if grid_visible:
        ax.grid(True, color='white')
    else:
        ax.grid(False)
    
    # Force immediate redraw
    fig.canvas.draw()

plt.show()

解决方案 #

问题核心在于两个细节：

1. 初始状态不匹配：代码默认开启了网格，但grid_visible初始设为False，导致第一次点击逻辑反转。
2. 网格样式丢失：重新开启网格时未指定完整样式，导致显示效果异常。

修改后的网格按钮代码如下：

ax_grid = fig.add_axes([0.85, 0.2, 0.1, 0.04])
gbutton = Button(ax_grid, 'Grid', color='0.3', hovercolor='0.7')
# 初始网格已开启，同步grid_visible状态
grid_visible = True

def grid_lines(event):
    global grid_visible
    grid_visible = not grid_visible
    # 切换网格状态时保留原有样式
    ax.grid(grid_visible, color='white', linestyle='--', linewidth=0.5, zorder=1)
    fig.canvas.draw_idle()

gbutton.on_clicked(grid_lines)

关键说明 #

• 初始grid_visible设为True，与代码中默认开启的网格状态保持一致，避免点击逻辑错误。
• 调用ax.grid()时带上完整样式参数，确保显示时保持白色虚线的原有样式。
• 使用fig.canvas.draw_idle()触发重绘，足够更新界面且性能更优，无需额外的plt.sca()或fig.canvas.draw()。

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