Aggiunta di frecce di flusso d'acqua a Matplotlib Contour Plot

Question

Sto generando un profilo altimetrico della falda con Matplotlib. Vedi sotto

Ecco quello che ho ora; come posso aggiungere frecce di flusso d'acqua come l'immagine qui sotto?

voglio aggiungere frecce per renderlo simile a questa:

Se qualcuno ha qualche idea e/o campioni di codice che sarebbe molto apprezzato.

Answer 1

Avrete bisogno di una versione recente (> = 1.2) di matplotlib, ma lo fa streamplot. Hai solo bisogno di prendere il gradiente negativo della tua testa (a.k.a. "tavola acquatica" per le falde acquifere di superficie).

Come esempio veloce generato dalle osservazioni punti casuali di testa:

import numpy as np 
from scipy.interpolate import Rbf 
import matplotlib.pyplot as plt 
# Make data repeatable 
np.random.seed(1981) 

# Generate some random wells with random head (water table) observations 
x, y, z = np.random.random((3, 10)) 

# Interpolate these onto a regular grid 
xi, yi = np.mgrid[0:1:100j, 0:1:100j] 
func = Rbf(x, y, z, function='linear') 
zi = func(xi, yi) 

# -- Plot -------------------------- 
fig, ax = plt.subplots() 

# Plot flowlines 
dy, dx = np.gradient(-zi.T) # Flow goes down gradient (thus -zi) 
ax.streamplot(xi[:,0], yi[0,:], dx, dy, color='0.8', density=2) 

# Contour gridded head observations 
contours = ax.contour(xi, yi, zi, linewidths=2) 
ax.clabel(contours) 

# Plot well locations 
ax.plot(x, y, 'ko') 

plt.show()