[python] 파이썬 랜덤 함수 (randint, choice, shuffle)

파이썬에는 랜덤 관련 다양한 함수가 존재합니다. 여기서는 주로 사용되는 몇가지 예를 알아보겠습니다. 간단한 예로 주어진 정수 범위 내에서 한 수를 랜덤하게 추출하는 예입니다.

a = np.random.randint(low=0, high=10, size=(2,5))
print(a)

위 코드는 0~10 사이의 정수들 중 10개를 뽑아 2X5 배열을 반환하게 됩니다. low, high, size 매개변수 모두 옵션입니다. 생략하고 아래와 같이 하나의 매개변수만 사용할 수 있습니다. 이때 매개변수로 전달되는 값은 양의 정수이며, 이 값을 N이라고 했을 때 0~N-1 사이의 임의의 정수 하나를 반환하게 됩니다.

a = np.random.randint(10)
print(a)

이미 만들어진 배열에서 임의의 항목을 추출하기 위해서는 choice 함수를 사용하면 됩니다.

b = np.random.choice([1, 3, 4, -6], size=(2,5))
print(b)

위에서 매개변수 size는 옵션입니다. 생략하는 경우 하나의 값을 반환합니다.

또한 주어진 배열의 내용을 섞고 싶은 경우 shuffle 함수를 사용하면 됩니다.

c = [1, 2, 3, 4]
for _ in range(3):
    np.random.shuffle(c)
    print(c)

끝.