如何使用高级索引独立滚动矩阵行？

使用高级索引独立滚动矩阵行

在数据操作领域，通常需要对多维数组执行复杂的操作。一种这样的场景涉及根据提供的移位值独立滚动矩阵的行。

给定一个输入矩阵 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中文网其他相关文章！