NumPy에서 고급 인덱싱을 사용하여 행렬 행의 독립적 롤링을 수행하는 방법은 무엇입니까?

고급 인덱싱을 사용하여 독립적으로 행렬 행 롤링

행렬 A와 각 행의 롤 값을 포함하는 배열 r이 주어지면 목표는 다음과 같습니다. 해당 롤 값을 사용하여 A의 각 행을 독립적으로 롤업합니다.

이를 달성하는 가장 효율적인 접근 방식은 NumPy의 고급 인덱싱을 이용하는 것입니다. 이 기술에는 A의 열에 롤 값을 적용하는 새로운 meshgrid를 구성하는 작업이 포함됩니다. 구현 방법은 다음과 같습니다.

<code class="python">import numpy as np

# Define the matrix A and roll values r
A = np.array([[4, 0, 0],
              [1, 2, 3],
              [0, 0, 5]])
r = np.array([2, 0, -1])

# Create a meshgrid of rows and negative shifted columns
rows, column_indices = np.ogrid[:A.shape[0], :A.shape[1]]
r[r < 0] += A.shape[1]
column_indices = column_indices - r[:, np.newaxis]

# Use advanced indexing to apply the roll values
result = A[rows, column_indices]

# Print the result
print(result)</code>

이 접근 방식은 음수로 이동된 열 인덱스를 사용하여 유효한 인덱싱을 보장하고 롤 값을 적용합니다. meshgrid를 통과하여 독립적으로 롤링된 행이 있는 행렬이 생성됩니다.

위 내용은 NumPy에서 고급 인덱싱을 사용하여 행렬 행의 독립적 롤링을 수행하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!