记录_List多索引

最新推荐文章于 2023-12-26 17:23:15 发布

Kougami..

最新推荐文章于 2023-12-26 17:23:15 发布

阅读量263

点赞数

文章标签：列表索引 numpy

本文链接：https://blog.csdn.net/weixin_44109545/article/details/115400426

版权

本文通过示例介绍如何在Python中处理列表的多级索引，包括example_1和example_2的详细步骤，进而推广到一般情况。讲解了如何遍历不同级别的索引并组合成新的数据结构，最后探讨了这种操作在从点云数据中选择特定部分的应用。

摘要由CSDN通过智能技术生成

遇到问题：
列表中的索引为一个列表
记录代入具体值试验的结果

example_1

points = torch.rand((5, 4, 3))
b_idx = torch.randint(4, (5, 2))
idx = torch.randint(4, (5, 2))
new = points[b_idx, idx, :]

points: 
>>>
tensor([[[5.9856e-01, 8.5678e-01, 9.8524e-01],
         [7.0864e-01, 7.1617e-01, 1.3145e-01],
         [1.3583e-01, 8.8172e-01, 1.0228e-01],
         [3.9284e-01, 5.1706e-01, 6.1809e-01]],

        [[7.0592e-01, 7.1845e-01, 9.4552e-01],
         [9.6846e-01, 8.6515e-01, 4.9927e-01],
         [8.3103e-01, 1.8320e-01, 5.3334e-01],
         [7.9671e-01, 9.9947e-01, 8.3414e-01]],

        [[5.1316e-01, 5.4100e-02, 9.3235e-01],
         [9.0255e-01, 4.5346e-01, 1.7189e-01],
         [8.1241e-01, 4.8314e-02, 5.3953e-01],
         [5.7105e-01, 1.1216e-01, 6.8769e-01]],

        [[4.5663e-01, 5.9473e-02, 7.5409e-01],
         [8.0928e-01, 3.8738e-01, 2.6865e-01],
         [9.4528e-01, 9.5299e-01, 2.9093e-01