在数据操作领域,通常需要对多维数组执行复杂的操作。一种这样的场景涉及根据提供的移位值独立滚动矩阵的行。
给定一个输入矩阵 A 和一组移位值 r,任务是将滚动操作应用于 A 的每一行使用 r 的相应移位。期望的结果是:
[[0 0 4] [1 2 3] [0 5 0]]
高级索引为这一挑战提供了一个优雅的解决方案。通过利用负移位值和高级数组切片技术,您可以有效地实现滚动操作,如下所示:
<code class="python">rows, column_indices = np.ogrid[:A.shape[0], :A.shape[1]] # Always use a negative shift, so that column_indices are valid. # Alternative: r %= A.shape[1] r[r < 0] += A.shape[1] column_indices = column_indices - r[:, np.newaxis] result = A[rows, column_indices]</code>
在这种方法中,ogrid 生成与 A 的行和列相对应的索引网格。根据负移位值操纵列索引,滚动操作有效地应用于每一行。该方法为独立滚动矩阵行提供了一种高效的解决方案,避免了循环的需要。
以上是如何使用高级索引独立滚动矩阵行?的详细内容。更多信息请关注PHP中文网其他相关文章!