numpy 学习笔记（备忘）

切片和索引

import numpy as np

a = np.random.randint(low=1, high=10, size=(3, 3, 2))
print(‘----------a----------\n‘, a)
print(‘-----a[:, 2, :]------\n‘, a[:, 2, :]) # 切片，第二维只取idx=2的，其他维全取
print(‘-----a[:][2][:]------\n‘, a[:][2][:]) # 索引，先a[:]=a，再a[:][2]=a[2]，再a[:][2][:]=a[2][:]=a[2]，所以就是a[2]

输出：

numpy 学习笔记（备忘）

原文：https://www.cnblogs.com/MrZJ/p/9755051.html