Numpy: операции над столбцами (или строками) массива NxM

Question

Это может быть глупый вопрос, но я только начал использовать numpy, и мне нужно выяснить, как выполнить некоторые простые операции.

Предположим, что у меня есть массив 2х3

array([[1, 3, 5], 
    [2, 4, 6]]) 

И что я хочу, чтобы выполнить какую-то операцию на первой колонке, например, вычесть 1 для всех элементов, чтобы получить

array([[0, 3, 5], 
    [1, 4, 6]]) 

Как может Я выполняю такую ​​операцию?

Answer 1

arr 
# array([[1, 3, 5], 
#  [2, 4, 6]]) 

arr[:,0] = arr[:,0] - 1  # choose the first column here, subtract one and 
          # assign it back to the same column 

arr 
# array([[0, 3, 5], 
#  [1, 4, 6]])