[3초 해결] 파이썬 np.random.randint 함수 범위 사용법

파이썬에서 무작위로 정수를 생성하되,

내가 범위를 지정해서(예를 들면 0~10 사이) 숫자도 만들고, 배열도 만들고 싶을 때 사용하는 함수다.

사용법이 아주 간단해서 3초 해결이라는 이름을 붙여보았다.

 

np.random.randint(low, high, size, dtype)

low에서 high 사이에 size만큼의 숫자를 생성한다. dtype은 따로 설정하지 않는다면 int.

 

유의해야 할 점이 몇 가지 있다.

1. low값은 생성할 숫자에 포함된다. 

2. high값은 생성할 숫자에 포함되지 않는다. 지정해주지 않으면 0부터 low 사이의 값이 만들어진다.

3. size에는 숫자가 들어갈 수도 있고, 튜플 형태로 배열을 지정해 줄 수도 있다.

 

import numpy as np
arr1 = np.random.randint(5) #0부터 5까지 랜덤하게 1개의 숫자
print("arr1 : ",arr1)
#>> arr1 :  1
arr2 = np.random.randint(5,10,3) #5부터 9까지 랜덤하게 3개의 숫자
print("arr2 : ",arr2)
#>> arr2 :  [5 7 7]
arr3 = np.random.randint(10, size=(2,3)) #2x3크기의 배열 생성, 배열의 값은 0~9 사이의 숫자.
print("arr3 : ",arr3)
#>>arr3 :  [[4 8 3],[5 0 2]]

실제 코드 실행 화면