如何在 numpy 中制作棋盘?

新手上路,请多包涵

我正在使用 numpy 将像素数组初始化为灰色棋盘(“无像素”或透明的经典表示)。似乎应该有一种奇妙的方法来使用 numpy 令人惊叹的数组赋值/切片/切块操作,但这是我想出的最好的方法:

 w, h = 600, 800
sq = 15    # width of each checker-square
self.pix = numpy.zeros((w, h, 3), dtype=numpy.uint8)
# Make a checkerboard
row = [[(0x99,0x99,0x99),(0xAA,0xAA,0xAA)][(i//sq)%2] for i in range(w)]
self.pix[[i for i in range(h) if (i//sq)%2 == 0]] = row
row = [[(0xAA,0xAA,0xAA),(0x99,0x99,0x99)][(i//sq)%2] for i in range(w)]
self.pix[[i for i in range(h) if (i//sq)%2 == 1]] = row

它有效,但我希望有更简单的东西。

原文由 Ned Batchelder 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 444
1 个回答
def checkerboard(shape):
    return np.indices(shape).sum(axis=0) % 2

最紧凑,可能是最快的,也是唯一发布的推广到 n 维的解决方案。

原文由 Eelco Hoogendoorn 发布,翻译遵循 CC BY-SA 4.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题