在步骤中提取指数

Question

我有一个numpy数组，如图所示，由红色和黄色像素组成。我只想选择红色的。

import numpy as np
data = np.ones((10, 10))

怎么做，伙计们？

Answer 1

好的，所以你似乎想要用交替/棋盘图案掩盖你的输入：

import numpy as np

def checkerboard(shape):
    "A hacky way to generate a checkerboard"
    return np.sum(np.indices(shape), axis=0) % 2 == 0

data = np.ones((10, 10), dtype=np.bool)

# equivalent ways of applying the mask to your array
result = data & checkerboard(data.shape)

# or 
result = np.logical_and(data, checkerboard(data.shape))

即。 checkerboard((5, 5))返回

array([[ True, False,  True, False,  True],
       [False,  True, False,  True, False],
       [ True, False,  True, False,  True],
       [False,  True, False,  True, False],
       [ True, False,  True, False,  True]], dtype=bool)