您如何遍历此数组？ 我知道这是非常基本的

Question

import numpy as np
a=np.arange(6)

for i in a:
    b[i]=a[i] + 1

print(b)

this is the error

IndexError: list assignment index out of range

I want b to read [1 2 3 4 5 6]

Answer 1

for i in array returns the values, not the indices. To iterate over the indices, use range(len(a)) instead of just a .

The corrected code:

for i in range(len(a)):
    b[i] = a[i] + 1

The better solution, if you're just looking for 1 to 6, is to do np.arange(1, 7)

Edit: As Paul points out, numpy has a much better solution than I knew. All you have to do is b = a + 1