使用两个范围对象切割秩2 numpy数组

作者: 一腔诗意喂了狗
发布时间: 2024-09-04 11:53:55 (1天前)
转自：

2 条回复

0#
回复此人
满目山河 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你为什么期望他们做同样的事情？ <code> numpy </code> 与这种索引不同于MATLAB。 </p> <pre> <code> In [6]: arr = np.arange(9).reshape(3,3) In [7]: arr[range(3),range(3)] Out[7]: array([0, 4, 8]) </code> </pre> <P> 使用范围（或相同值的列表或数组），它返回1d数组，在这种情况下是2d数组的对角线。此索引选择一组点，而不是块。 </p> <P> 在 <code> numpy </code> 如果你想要一个带有'高级索引'的2d结果，你需要创建一对索引数组，它们一起广播到正确的形状： </p> <pre> <code> In [8]: arr[np.arange(3)[:,None],range(3)] Out[8]: array([[0, 1, 2], [3, 4, 5], [6, 7, 8]]) </code> </pre> <P> 在MATLAB / octave中，很容易选择一个块，但更难选择单个元素。 </p> <pre> <code> arr(sub2ind([3,3],[1,2,3],[1,2,3])) </code> </pre> <P> 一般来说，要了解 <code> numpy </code> 您需要了解广播的高级索引 - 适用于索引以及添加等数学运算。 </p> </DIV>

编辑

登录后才能参与评论