一维数组的索引、切片和迭代等操作与Python的列表类似,示例代码如下。
rng = np.random.default_rng() #获得随机数生成器
a=rng.integers(10,size=8)
a
array([6, 1, 4, 6, 0, 0, 1, 0], dtype=int64)
a[0] #索引:第1个元素
a[-1] #索引:最后1个元素
a[2] #索引:第3个元素
a[2:5] #切片
array([4, 6, 0], dtype=int64)
a[:2]
array([6, 1], dtype=int64)
a[5:]
array([0, 1, 0], dtype=int64)
for x in a: #迭代
... print(x,end=' ')
多维数组用以逗号分隔的多个值进行索引,示例代码如下。
rng = np.random.default_rng()
a=rng.integers(10,size=(2,5)) #创建一个大小为2×5的数组,元素为10以内的随机整数
a
array([[6, 8, 9, 5, 7],
[8, 3, 1, 2, 1]], dtype=int64)
a[0,0] #索引:第1行第1个元素
6
a[1,0] #索引:第2行第1个元素
8
a[0,:3] #切片:第1行前3个元素
array([6, 8, 9], dtype=int64)
for x in a: #迭代
... print(x)