大蟒蛇python教程共享如何利用python绘制等高线图

matplotlib.pyplot.contour(*args, data=none, **kwargs)

import numpy as np  import matplotlib.pyplot as plt    x, y = np.meshgrid(np.linspace(-3,3,256), np.linspace(-3,3,256))  z = (1 - x/2 + x**5 + y**3) * np.exp(-x**2 - y**2)  levels = np.linspace(np.min(z), np.max(z), 7)  fig, ax = plt.subplots()  ax.contour(x, y, z, levels=levels)  plt.show()

import numpy as np  import matplotlib.cm as cm  import matplotlib.pyplot as plt    delta = 0.025  x = np.arange(-3.0, 3.0, delta)  y = np.arange(-2.0, 2.0, delta)  x, y = np.meshgrid(x, y)  z1 = np.exp(-x**2 - y**2)  z2 = np.exp(-(x - 1)**2 - (y - 1)**2)  z = (z1 - z2) * 2  fig, ax = plt.subplots()  cs = ax.contour(x, y, z)  ax.clabel(cs, inline=true, fontsize=10)  ax.set_title('simplest default with labels')  plt.show()

import numpy as np  import matplotlib.cm as cm  import matplotlib.pyplot as plt    delta = 0.025  x = np.arange(-3.0, 3.0, delta)  y = np.arange(-2.0, 2.0, delta)  x, y = np.meshgrid(x, y)  z1 = np.exp(-x**2 - y**2)  z2 = np.exp(-(x - 1)**2 - (y - 1)**2)  z = (z1 - z2) * 2  fig, ax = plt.subplots()  cs = ax.contour(x, y, z, 6, colors='k')  ax.clabel(cs, fontsize=9, inline=true)  ax.set_title('single color - negative contours dashed')  plt.show()

import numpy as np  import matplotlib.cm as cm  import matplotlib.pyplot as plt    delta = 0.025  x = np.arange(-3.0, 3.0, delta)  y = np.arange(-2.0, 2.0, delta)  x, y = np.meshgrid(x, y)  z1 = np.exp(-x**2 - y**2)  z2 = np.exp(-(x - 1)**2 - (y - 1)**2)  z = (z1 - z2) * 2  fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(14,7))  axs[0].set_title('levels=6')  cs = axs[0].contour(x, y, z, 6, colors='k')  axs[0].clabel(cs, fontsize=9, inline=true)    axs[1].set_title('levels=10')  cs1 = axs[1].contour(x, y, z, 10, colors='k')  axs[1].clabel(cs1, fontsize=9, inline=true)  plt.show()

import numpy as np  import matplotlib.cm as cm  import matplotlib.pyplot as plt    delta = 0.025  x = np.arange(-3.0, 3.0, delta)  y = np.arange(-2.0, 2.0, delta)  x, y = np.meshgrid(x, y)  z1 = np.exp(-x**2 - y**2)  z2 = np.exp(-(x - 1)**2 - (y - 1)**2)  z = (z1 - z2) * 2    fig, ax = plt.subplots()  cs = ax.contour(x, y, z, 6,                  linewidths=np.arange(.5, 4, .5),                  colors=('r', 'cyan', 'blue', (1, 1, 0), '#afeeee', '0.5'),                  )  ax.clabel(cs, fontsize=9, inline=true)  ax.set_title('crazy lines')  plt.show()

import numpy as np  import matplotlib.cm as cm  import matplotlib.pyplot as plt    delta = 0.025  x = np.arange(-3.0, 3.0, delta)  y = np.arange(-2.0, 2.0, delta)  x, y = np.meshgrid(x, y)  z1 = np.exp(-x**2 - y**2)  z2 = np.exp(-(x - 1)**2 - (y - 1)**2)  z = (z1 - z2) * 2  fig, ax = plt.subplots()  im = ax.imshow(z, interpolation='bilinear', origin='lower',                 cmap=cm.gray, extent=(-3, 3, -2, 2))  levels = np.arange(-1.2, 1.6, 0.2)  cs = ax.contour(z, levels, origin='lower', cmap='flag', extend='both',                  linewidths=2, extent=(-3, 3, -2, 2))  cs.collections[6].set_linewidth(4)  ax.clabel(cs, levels[1::2],  # label every second level            inline=true, fmt='%1.1f', fontsize=14)  cb = fig.colorbar(cs, shrink=0.8)  ax.set_title('lines with colorbar')  cbi = fig.colorbar(im, orientation='horizontal', shrink=0.8)  l, b, w, h = ax.get_position().bounds  ll, bb, ww, hh = cb.ax.get_position().bounds  cb.ax.set_position([ll, b + 0.1*h, ww, h*0.8])  plt.show()