numpy数组索引与切片

import numpy as np

# NumPy支持强大的数组索引和切片功能
# 随机生成一维数组
# low、high分别表示随机整数的下界和上界，size表示数组大小
a = np.random.randint(low=1, high=20, size=5)
print(a)

# 获取一维数组a中索引为2~3的元素
print(a[2:4])

# 随机生成二维数组
a = np.random.randint(low=1, high=20, size=(5,5))
print(a)

# 第2~4行中的第2列元素（此处行列均为索引）
print(a[2:5, 2])

# a中的1~2行，输出所有列
print(a[1:3, ])

# 对a中1~3行、2~3列进行切片并赋值给b
b = a[1:4, 2:4]
print(b)

# 还可以通过 reshape(）函数改变数组形状
# 改变数组形状
a = np.random.randint(low=1, high=20, size=9)
print(a)
a = a.reshape(3,3)
print(a)