>백엔드 개발 >파이썬 튜토리얼 >NumPy에서 특정 스트라이드를 사용하여 하위 배열을 효율적으로 추출하는 방법은 무엇입니까?

NumPy에서 특정 스트라이드를 사용하여 하위 배열을 효율적으로 추출하는 방법은 무엇입니까?

Susan Sarandon
Susan Sarandon원래의
2024-12-06 09:41:11813검색

How to Efficiently Extract Subarrays with a Specific Stride in NumPy?

NumPy를 사용하여 스트라이드로 하위 배열 추출

NumPy 배열이 제공되면 특정 스트라이드를 사용하여 하위 배열로 분할해야 하는 경우가 많습니다. 이 질문은 브로드캐스팅과 NumPy 스트라이드를 사용하여 이를 효율적으로 달성하는 방법을 탐구합니다.

해결책 1: 브로드캐스팅

Broadcast_app 함수는 브로드캐스팅을 사용하여 원하는 하위 배열을 구성합니다. 배열 크기, 하위 배열 길이 및 보폭을 기준으로 행 수를 계산합니다. 그런 다음 브로드캐스팅을 사용하여 각 행이 하위 배열을 나타내는 새 배열을 생성합니다.

해결책 2: NumPy Strides

strided_app 함수는 NumPy의 효율적인 보폭 처리 메커니즘을 활용합니다. 이전과 마찬가지로 행 수와 보폭을 계산합니다. 그런 다음 as_strided 함수를 활용하여 원하는 보폭과 모양을 가진 새 배열을 생성합니다.

샘플 사용법

이러한 솔루션을 설명하려면 다음 Python 코드를 고려하세요.

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
L = 5  # Subarray length
S = 3  # Stride

print(broadcasting_app(a, L, S))
print(strided_app(a, L, S))

출력:

[[ 1  2  3  4  5]
 [ 4  5  6  7  8]
 [ 7  8  9 10 11]]
[[ 1  2  3  4  5]
 [ 4  5  6  7  8]
 [ 7  8  9 10 11]]

둘 다 접근 방식은 원하는 하위 배열 행렬을 효율적으로 생성합니다.

위 내용은 NumPy에서 특정 스트라이드를 사용하여 하위 배열을 효율적으로 추출하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.