[Python] np.arange

▶ np.arange

• Numpy 라이브러리에서 제공하는 함수
• 지정된 범위 내에서 일정 간격으로 배열을 생성하는 데 사용
• Python의 range와 비슷하지만, 정수 뿐만 아니라 부동소수점 간격도 지원한다는 점에서 더 유용함

1. 사용법

numpy.arange([start, ]stop, [step, ]dtype=None, *, like=None)

• start (optional)
  • 생성할 배열의 시작 값 지정
  • 기본값 = 0
• stop
  • 배열 생성이 멈출 값을 지정
  • stop은 포함되지 않음
• step (optional)
  • 값들 간의 간격 지정
  • 기본값 = 1
• dtype (optional)
  • 생성된 배열의 데이터 타입 지정
  • 지정하지 않으면 입력값에 따라 자동 추론
• like (optional)
  • 배열 생성 시 참조할 객체 지정
  • 주로 Numpy API 호환성을 위해 사용

2. 반환값

• 지정된 범위와 간격에 따라 생성된 Numpy 배열 반환

3. 예제

• 정수 간격으로 배열 생성

import numpy as np
arr = np.arange(5)
print(arr)  # 출력: [0 1 2 3 4]

• start와 stop 지정

arr = np.arange(2, 8)
print(arr)  # 출력: [2 3 4 5 6 7]

• 소수점 간격

arr = np.arange(0, 2, 0.5)
print(arr)  # 출력: [0.  0.5  1.  1.5]

• dtype 지정

arr = np.arange(1, 5, dtype=float)
print(arr)  # 출력: [1. 2. 3. 4.]

4. 주의사항

• 소수점 간격 사용 시, 부동소수점 오차로 인해 예상과 약간 다른 결과가 나올 수 있음
  → 필요 시 np.linspace를 사용하는 것이 더 나을 수도 있음