如何在同一个numpy数组上使用matplotlib的imshow和等高线图？

Question

使用以下代码，我在2D笛卡尔网格上生成障碍物（稍后用作某些模拟的输入），方法是将障碍物所在的区域定义为1和0。

我想用等高线图绘制它，但有一些麻烦产生二进制填充轮廓图（侧面问题：如何实现？）因此决定将数组绘制为图像。

以下是代码：

import numpy as np, matplotlib.pyplot as plt

# create spatial coordinates
N_x, N_y = 161, 161
x = np.linspace( .0, .80, N_x )
y = np.linspace( .0, .80, N_y )

# define obstacle
obst_diameter = .3
obst_center   = (.4,.2)

# 2D array defining obstacle structure: 1=obstacle, 0=nothing
metal = np.zeros( (N_x, N_y), dtype=int )

for ii in range(N_x):
    for jj in range(N_y):
        if (     x[ii] >= (obst_center[0]-obst_diameter/2)
             and x[ii] <= (obst_center[0]+obst_diameter/2)
             and y[jj] >= (obst_center[1]-obst_diameter/2)
             and y[jj] <= (obst_center[1]+obst_diameter/2)
           ):
            metal[jj,ii] = 1.

# do the plotting (contour and imshow)
xx, yy = np.meshgrid( x, y )
fig = plt.figure( figsize=(20,10) )

ax1 = fig.add_subplot( 1,2,1, aspect='equal' )
cont_obst = ax1.contour( xx, yy, metal, colors='k', levels=[0,1] )
ax1.plot( obst_center[0], obst_center[1], marker='*', markersize=30, color='yellow' )
ax1.set_xlabel( 'x in m' )
ax1.set_ylabel( 'y in m' )

ax2 = fig.add_subplot( 1,2, 2, aspect='equal' )
ax2.imshow( metal, cmap='Greys', interpolation='none',
        extent=[np.min(x), np.max(x), np.min(y), np.max(y)] )
ax2.plot( obst_center[0], obst_center[1], marker='*', markersize=30, color='yellow' )
ax2.set_xlabel( 'x in m' )
ax2.set_ylabel( 'y in m' )

plt.savefig( 'another_plot.png', bbox_inches='tight' )

生成的图像如下所示，左侧为contour图，右侧为imshow图（障碍物中心标有黄色星）。

显然，这两个地块是不同的。这是什么原因，即我在这里缺少什么？

Answer 1

matplotlib的 imshow 功能的原点是左上角。如果您将相关行更改为：

ax2.imshow( metal, cmap='Greys', interpolation='none',
        extent=[np.min(x), np.max(x), np.min(y), np.max(y)], origin='lower')

它将解决此问题。