如何对 numpy 数组中的第一个索引求和

问题描述 投票:0回答:1

如何将数组的第一个索引与迭代次数相加,最大迭代次数为 2.

我已经尝试使用下面的代码,但我仍然得到索引号 2 的错误结果。

a = np.array([[858,833,123],
[323,542,927],
[938,361,271],
[679,272,451]])

max_iter = 2
for i in range(max_iter):
    for j in range(len(a)):
        a[j][0] += i

上面代码的结果:

[[859 833 123]
 [324 542 927]
 [939 361 271]
 [680 272 451]]

我的预期结果是:

[859,833,123],
[325,542,927],
[939,361,271],
[681,272,451]
python
1个回答
0
投票

1
2
值分别添加到偶数和奇数索引行(第一列):

a[::2, 0] += 1
a[1::2, 0] += 2

array([[859, 833, 123],
       [325, 542, 927],
       [939, 361, 271],
       [681, 272, 451]])
© www.soinside.com 2019 - 2024. All rights reserved.