замаскируйте двумерный массив numpy на основе значений в одном столбце

Предположим, у меня есть следующий массив numpy:

a = [[1, 5, 6],
     [2, 4, 1],
     [3, 1, 5]]

Я хочу замаскировать все строки, которые имеют ] 1 в первом столбце. То есть, я хочу

   [[--, --, --],
     [2, 4, 1],
     [3, 1, 5]]

Можно ли это сделать, используя множество операций с маскированными массивами? Как это сделать?

Спасибо.

11
задан Curious2learn 7 January 2011 в 13:08
поделиться