如何将数组的第一个索引与迭代次数相加,最大迭代次数为 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]
将
1
和2
值分别添加到偶数和奇数索引行(第一列):
a[::2, 0] += 1
a[1::2, 0] += 2
array([[859, 833, 123],
[325, 542, 927],
[939, 361, 271],
[681, 272, 451]])