사용자가 입력한 범위 내에서 NumPy 배열을 생성하는 방법

사용자가 입력한 범위 안에서 NumPy 배열을 생성하는 방법을 알아보겠습니다. 이때 NumPy 라이브러리에서 제공하는 arange() 함수를 사용하면 간단하게 원하는 결과를 얻을 수 있습니다.

arange() 함수란?

np.arange(start, stop, step)은 시작값(start)부터 끝값(stop) 직전까지, 지정한 간격(step)만큼 떨어진 숫자들로 이루어진 배열을 반환합니다. 단, 끝값 자체는 배열에 포함되지 않는다는 점에 유의해야 합니다.

알고리즘

1단계: NumPy를 임포트합니다.
2단계: 사용자로부터 시작값, 끝값, 간격(스텝)을 입력받습니다.
3단계: NumPy의 arange() 함수를 사용해 배열을 생성하고 출력합니다.

예제 코드

import numpy as np

start_val = int(input("Enter starting value: "))
end_val = int(input("Enter ending value: "))
step_val = int(input("Enter step value: "))
print(np.arange(start_val, end_val, step_val))

실행 결과

Enter starting value: 5
Enter ending value: 50
Enter step value: 5
[ 5 10 15 20 25 30 35 40 45]

실행 결과를 보면 시작값 5부터 끝값 50 미만까지 5씩 증가하는 숫자들이 배열로 출력됩니다. 끝값인 50은 범위에 포함되지 않기 때문에 배열의 마지막 요소는 45가 됩니다. 이처럼 arange() 함수를 활용하면 사용자가 원하는 시작값, 끝값, 간격을 자유롭게 지정하여 배열을 손쉽게 만들 수 있습니다.