콘텐츠로 건너뛰기

Numpy arange 함수의 사용법 및 수정 방법: 어떻게 사용하나요?

CodeMDD.io

NumPy arange(): np.arange() 사용 방법

NumPy arange() 함수는 숫자 범위에 기반한 배열 생성 함수입니다. 이 함수는 균등하게 간격이 떨어진 값들로 이루어진 ndarray 인스턴스를 생성하고 참조를 반환합니다. NumPy에서 배열을 만들고 사용하는 것은 SciPy, Pandas, Matplotlib, scikit-learn 등과 같은 다른 Python 라이브러리와 함께 작업할 때 중요합니다. NumPy는 유용한 루틴을 제공하며 성능 향상을 가능하게 해주며 간결한 코드 작성이 가능합니다.

이 글의 끝까지 가면 다음과 같은 내용을 배울 수 있습니다:

  • np.arange()가 무엇인지
  • np.arange()를 어떻게 사용하는지
  • np.arange()가 Python 내장 클래스인 range와 비교해 어떤 차이가 있는지
  • np.arange()와 비슷한 루틴들은 무엇인지

이제 np.arange()를 실제로 살펴보겠습니다!

np.arange()의 반환 값과 매개 변수

NumPy의 arange() 함수는 숫자 범위를 기반으로한 배열 생성 함수 중 하나입니다. 이 함수는 균등하게 간격이 떨어진 값들로 이루어진 ndarray 인스턴스를 생성하고 그 참조를 반환합니다.

arange() 함수는 다음과 같은 네 가지 매개 변수로 값의 범위, 간격, 그리고 타입을 정의할 수 있습니다.

Python

numpy.arange([start, ]stop, [step, ], dtype=None) -> numpy.ndarray

첫 세 개의 매개 변수는 값의 범위를 결정하고, 네 번째 매개 변수는 배열의 요소 타입을 지정합니다.

  1. start는 배열의 첫 번째 값으로 정의되는 숫자(정수 또는 실수)입니다.
  2. stop은 배열의 끝을 정의하는 숫자로, 배열에 포함되지 않습니다.
  3. step은 배열의 연속된 두 값 사이의 간격(차이)을 정의하는 숫자로, 기본값은 1입니다.
  4. dtype은 출력 배열의 요소 타입으로, 기본값은 None입니다.

step은 0이 될 수 없습니다. 그렇지 않으면 ZeroDivisionError가 발생합니다. 만약 increment 또는 decrement가 0이면 시작점에서 움직임이 없게 됩니다.

dtype이 생략되면, 정의되지 않은 경우 자동으로 추론됩니다.

import numpy as np
arr = np.arange(5) # 0부터 4까지의 값을 갖는 배열 생성
print(arr) # 출력: [0 1 2 3 4]
arr = np.arange(2, 9, 3) # 2부터 시작하여 3씩 증가하는 값으로 배열 생성
print(arr) # 출력: [2 5 8]
arr = np.arange(5, dtype=float) # 0부터 4까지의 값을 float 타입으로 갖는 배열 생성
print(arr) # 출력: [0. 1. 2. 3. 4.]

np.arange() 함수를 사용하여 간편하게 배열을 생성할 수 있고, 여러 다양한 용도로 활용할 수 있습니다. NumPy의 배열 생성 함수들을 활용하면 데이터 분석 및 과학적인 연구에 매우 효과적으로 사용할 수 있습니다.