Plotten Richtungsfeld in Python

Question

Ich möchte für eine einfache Gleichung Richtungsfeld zeichnen:

y' = 3 − 2y 

Ich habe ähnliches Matlab Problem here (1.3) gefunden. Aber ich weiß nicht, wie ich es in Python umschreiben soll. Mein letzter Versuch ist:

from matplotlib.pyplot import cm 
import matplotlib.pyplot as plt 
import numpy as np 

nx, ny = .3, .3 
x = np.arange(-3, 3, nx) 
y = np.arange(-2, 2, ny) 
X, Y = np.meshgrid(x, y) 

dy = X + np.sin(Y) 
dx = np.ones((10,10)) 

plot2 = plt.figure() 
plt.quiver(X, Y, dx, dy, 
      color='Teal', 
      headlength=7) 

plt.title('Quiver Plot, Single Colour') 
plt.show(plot2) 

Aber ich bin immer Fehler:

builtins.ValueError: operands could not be broadcast together with shapes (100,) (280,) 

ich, obwohl es sehr einfach sein, aber nach wenigen Stunden der Suche, wie man ein einfaches Richtungsfeld zu zeichnen, ich bin ziemlich deprimiert.

Answer 1

dx und dy müssen die gleiche Form haben wie X und Y.

Derzeit haben Sie eine Form von (14, 20) für X, Y und dy, aber (10,10) für dx.

Wenn Sie die Zeile definieren dx zu ändern:

dx = np.ones(dy.shape) 

Alles funktioniert gut:

Answer 2

Sie auch die streamlines des eingereichten nutzen könnten einen schönen Eindruck von der Strömung geben , und färbt die Kurven gemäß einer Eigenschaft des Feldes (dy in diesem Fall). Schauen Sie sich das folgende Beispiel:

nx, ny = .3, .3 
x = np.arange(-3, 3, nx) 
y = np.arange(-2, 2, ny) 
X, Y = np.meshgrid(x, y) 
dy = X + np.sin(Y) 
dx = np.ones(dy.shape) 

color = dy 
lw = 1 
plt.streamplot(X,Y,dx, dy, color=color, density=1., cmap='jet', arrowsize=1) 

, die produziert: