Возможно, вы можете попробовать это:
import numpy as np
q = np.ones((2, 1))
q[your_index_function(args)] = 0
и your_index_function(args)
должны возвращать тот же массив фигур, что и q
с логическими значениями в соответствии с вашими условиями (например, поведение idx_MV
?)