多维数组的索引与切片_numpy,pytorch等框架的高级索引

本文介绍了在numpy和pytorch等框架中,如何进行高级索引,包括整数数组索引、布尔索引和花式索引。通过具体示例展示了如何精确选取和操作多维数组中的特定元素,例如改变矩阵对角线元素、筛选符合条件的值,以及创建多头组合矩阵等。
摘要由CSDN通过智能技术生成

9fc75e86d91bce7f6265894ddc784d26.png

高级索引

在numpy或pytorch等框架中的对张量的操作不只提供了类似python列表的切片索引等操作,还提供了高级索引。

高级索引大致可分为整数数组索引、布尔索引及花式索引三类:

整数数组索引

整数数组索引指使用同形状的多个数组分别指定元素的所有维度(不指定的维度也可以用切片:或省略号...与索引数组组合),可精准取出一批指定位置的元素,按照给定数组形状返回。

如给定4*3 的二维数组

x = np.arange(12).reshape(4,3)
print(x)

输出为:

[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]]

1.取出其(0,0),(1,1)和(2,0)位置处的元素。

给出两个一维数组分别指定第0维和第1维

y = x[[0,1,2],  [0,1,0]]
print(y)
#[0 4 6]

2.传入的位置指定数组也可为多维,取出其[[(0,0)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值