![](/img/trans.png)
[英]How to plot 2 maps in 1 plot using Matplotlib? (Python)
[英]How to plot using matplotlib (python) colah's deformed grid?
我需要在 Python 中创建一个可视化,就像 colah 在他的网站上所做的那样。 但是,我无法在 matplotlib 上找到网格的任何失真以完全像他在这里所做的那样执行。 请帮助我,如果可以的话。
我猜图像是通过向网格添加一些高斯函数产生的。
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
def plot_grid(x,y, ax=None, **kwargs):
ax = ax or plt.gca()
segs1 = np.stack((x,y), axis=2)
segs2 = segs1.transpose(1,0,2)
ax.add_collection(LineCollection(segs1, **kwargs))
ax.add_collection(LineCollection(segs2, **kwargs))
ax.autoscale()
f = lambda x,y : ( x+0.8*np.exp(-x**2-y**2),y )
fig, ax = plt.subplots()
grid_x,grid_y = np.meshgrid(np.linspace(-3,3,20),np.linspace(-3,3,20))
plot_grid(grid_x,grid_y, ax=ax, color="lightgrey")
distx, disty = f(grid_x,grid_y)
plot_grid(distx, disty, ax=ax, color="C0")
plt.show()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.