1

我正在尝试使用 matplotlib 库和 python 3 将颜色映射应用于二维数组。为简单起见，我制作了一个示例代码来显示我的问题：

示例代码：

    from matplotlib import pyplot as plt
    from matplotlib import colors
    import numpy as np

    #just an example array
    sample= np.zeros((20,20),dtype=int)
    sample[2,2]=sample[4,4]=2
    #color-table and color bounds
    cmap = colors.ListedColormap(['black','white','red','green'])
    bounds=[-6,0,1,99,105]
    norm = colors.BoundaryNorm(bounds, cmap.N)

    fig, ax = plt.subplots()
    ax.imshow(sample,interpolation='nearest', cmap=cmap, norm=norm)
    ax.grid(which='major', axis='both', linestyle='-', color='k', linewidth=2)
    ax.set_xlim(0, 20)
    ax.set_ylim(20, 0)

    ax.set_xticks(np.arange(0,20,1))
    ax.set_yticks(np.arange(0,20,1))
    plt.show()

使用这段代码，我有一个 20 x 20 的网格，除了两个单元格之外，其他地方都是白色的[2,2]，[4,4]应该是红色的。问题是当我显示我的网格时，单元格在两个方向上都偏离了 -0.5。我从这个示例代码中得到的是： 在此处输入图像描述

如果我将网格移动-0.5：

    ax.set_xticks(np.arange(-0.5,20,1))
    ax.set_yticks(np.arange(-0.5,20,1))

然后我的单元格看起来正确，现在网格都错了！ 在此处输入图像描述

我的代码有什么问题？我真的不明白这怎么会发生？！

4

1 回答 1

2

文档字符串的注释部分ax.imshow说：

除非使用范围，否则像素中心将位于整数坐标处。换句话说：原点将与像素 (0, 0) 的中心重合。

所以你的电话变成：

ax.imshow(sample, interpolation='none', cmap=cmap, norm=norm,
          extent=[0, 20, 20, 0])
于 2018-04-26T13:19:45.533 回答