python - Numpy:对 NxM 数组的列(或行)的操作

Question

这可能是一个愚蠢的问题，但我刚刚开始使用 numpy，我必须弄清楚如何执行一些简单的操作。

假设我有 2x3 数组

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]])