условие numpy.where для выбора определенных вертикальных столбцов в 2D массиве

Question

скажем, у нас есть

a = numpy.arange(25).reshape(5,5)

> array([[ 0,  1,  2,  3,  4],
         [ 5,  6,  7,  8,  9],
         [10, 11, 12, 13, 14],
         [15, 16, 17, 18, 19],
         [20, 21, 22, 23, 24]])

Идя

numpy.where(a[1])

> array([0, 1, 2, 3, 4])

и затем что-то вроде

a[1][numpy.where(a[1])]

> array([5, 6, 7, 8, 9])

Я могу выбрать горизонтальные строки массива исоответствующие значения, однако, как я могу иметь подобное условие, где выбрать только конкретные вертикальные столбцы

т.е..

numpy.where(condition)

> array([1, 6, 11, 16, 21]) 

Answer 1

Загляните в этот учебник, чтобы узнать, как применять нарезку на массивах numpy (https://docs.scipy.org/doc/numpy-1.15.1/reference/arrays.indexing.html).. На ваш вопрос ответ:

a[:,1]

Answer 2

Я точно не уверен, что это именно то, что вы имеете в виду, но вы можете индексировать столбцы, используя [:,column_number], где : обозначает «все строки»:

a[:,1][numpy.where(a[1])]

# array([ 1,  6, 11, 16, 21])

Выше, однако, эквивалентно просто a[:,1]:

>>> a[:,1]
array([ 1,  6, 11, 16, 21])