Я понял, как применять маску в столбцах.
import numpy as np
import random
X = np.arange(12).reshape((3, 4))
mask = np.array([1, 0, 1, 0], dtype=bool)
row = np.array([0, 1, 2])
X[row[:, np.newaxis], mask]
Но когда я попытался применить маску в столбце, я получаю сообщение об ошибке.
mask = np.array([[1],[0],[1]],dtype=bool)
X[row[:, np.newaxis],mask]