Anzeige zwei Sympy Plots als zwei Matplotlib Nebenhandlungen

Question

Dieser Code

from sympy import * 
x=Symbol('x') 
p1 = plot(x**2,(x,-2,2)) 
p2 = plot(x**3,(x,-2,2)) 

Ergebnisse in zwei getrennten Parzellen.

Anstelle von zwei separaten Grundstücken, ich möchte, dass sie als Nebenhandlungen mit matplotlib anzuzeigen:

import matplotlib.pyplot as plt 
fig = plt.figure() 
ax1 = fig.add_subplot(121) 
ax2 = fig.add_subplot(122) 
plt.show() 

Wie kann ich p1 und p2, hinzufügen, so dass sie als Nebenhandlungen innerhalb der matplotlib Figur dargestellt?

Answer 1

Das Problem ist, dass sympy Plot seine eigene Figur und Achsen erstellt. Es ist nicht beabsichtigt, auf eine bestehende Achse zu zeichnen.

Sie können jedoch die Achsen, auf denen das Diagramm gezeichnet wird, durch vorhandene Achsen ersetzen, bevor Sie das Symplekeil zeigen.

from sympy import Symbol,plot 
import matplotlib.pyplot as plt 

def move_sympyplot_to_axes(p, ax): 
    backend = p.backend(p) 
    backend.ax = ax 
    backend.process_series() 
    backend.ax.spines['right'].set_color('none') 
    backend.ax.spines['bottom'].set_position('zero') 
    backend.ax.spines['top'].set_color('none') 
    plt.close(backend.fig) 


x=Symbol('x') 
p1 = plot(x**2,(x,-2,2), show=False) 
p2 = plot(x**3,(x,-2,2), show=False) 


fig, (ax,ax2) = plt.subplots(ncols=2) 
move_sympyplot_to_axes(p1, ax) 
move_sympyplot_to_axes(p2, ax2) 

plt.show() 

Answer 2

Meine Lösung nicht hinzufügen p1, p2 die direkt Nebenhandlungen. Aber (x, y) Koordinaten von ihnen werden erfasst und stattdessen verwendet.

import matplotlib.pyplot as plt 
from sympy import symbols 
import numpy as np 

from sympy import symbols 
from sympy.plotting import plot 

# part 1 
# uses symbolic plot of functions 
x = symbols('x') 

#p1, p2 = plot(x**2, x**3, (x, -2, 2)) 

# this plot will not show ... 
# only produce 2 curves 
p1, p2 = plot((x**2, (x, -2, 2)), \ 
       (x**3, (x, -2, 2)), \ 
       show=False) 

# collect (x,y)'s of the unseen curves 
x1y1 = p1.get_points() # array of 2D 
x2y2 = p2.get_points() 

# part 2 
# uses regular matplotlib to plot the data 

fig = plt.figure(figsize=(8, 5)) 
ax1 = fig.add_subplot(121) 
ax2 = fig.add_subplot(122) 

# do subplot 1 
ax1.plot(x1y1[0], x1y1[1], 'g') # plot x**2 in green 
ax1.set_xlim([-2, 2]) 
ax1.set_xlabel('X1') 
ax1.set_ylabel('Y1') 
ax1.set_title('Line1') # destroyed by another .title(); axis metho1 

# do subplot 2 
ax2.plot(x2y2[0], x2y2[1], 'r') # plot x**3 in red 
ax2.set_xlim([-2, 2]) 
ax2.set_xlabel('X2') 
ax2.set_ylabel('Y2') 
ax2.set_title('Line2') 

fig.subplots_adjust(wspace=0.4) # set space between subplots 

plt.show() 

Das resultierende Grundstück: