如何在 matplotlib 中用 pcolor 绘制四边形补丁？

Question

我遇到了来自 matplotlib 的pcolor()问题。 我想绘制具有四边形形状的补丁。 （它们确实具有非规则形状，但确实有 4 个角并且不重叠）。 为了演示我的问题，这是一个产生意外行为的最小代码。

import numpy as np
from matplotlib import pyplot as plt

x = [0, 1]
y = [0, 2]
val = [[1]]
xx, yy = np.meshgrid(x, y)

fig, ax = plt.subplots(1, 2, figsize=[10, 10])

ax = plt.subplot(2, 1, 1)
yy[1,0] =2.9
#displays a 1x2 rectangle
ax.pcolor(xx, yy, val, edgecolors='black')

ax = plt.subplot(2, 1, 2)
yy[1,0] =3
#displays the expected 1x3(x2) trapezoid
ax.pcolor(xx, yy, val, edgecolors='black')

plt.show()

（我们期待一个高度为 2.9 的梯形，但它输出一个 2x1 的矩形。高度为 3 的第二个例子工作正常）

这是输出

此外，使用pcolormesh ，也会出现同样的问题。

这是一个错误还是我只是瞎了眼？

Answer 1

问题是xx和yy的数据类型是int64 。 因此，当您尝试赋值yy[1, 0] = 2.9它会被转换为int ，从而导致yy[1, 0] == 2 。

解决方案是将数据类型更改为float ，最简单的方法是使用numpy.ndarray.astype ：

import numpy as np
from matplotlib import pyplot as plt

x = [0, 1]
y = [0, 2]
val = [[1]]
xx, yy = np.meshgrid(x, y)

xx = xx.astype('float64')
yy = yy.astype('float64')

fig, ax = plt.subplots(1, 2, figsize=[10, 10])

ax = plt.subplot(2, 1, 1)
yy[1,0] = 2.9
ax.pcolor(xx, yy, val, edgecolors='black')

ax = plt.subplot(2, 1, 2)
yy[1,0] = 3
ax.pcolor(xx, yy, val, edgecolors='black')

这将给出预期的结果