如何同时修改n列numpy数组?

时间:2016-05-19 10:46:14

标签: python arrays python-2.7 numpy

如何修改n列numpy数组?例如,我有一个numpy数组如下:

P  = array([[1, 2, 3, 8, 6],
            [4, 5, 6, 4, 5]
            [0,-2, 5, 3, 0]])

如何更改P的第一列,第二列和第四列的元素?

1 个答案:

答案 0 :(得分:3)

使用indexing

以下是一个例子:

>>> P[:, [0, 1, 3]] += 10
>>> 
>>> P
array([[11, 12,  3, 18,  6],
       [14, 15,  6, 14,  5],
       [10,  8,  5, 13,  0]])