Mô phỏng để tìm xác suất

#   _______________________________________________________________________
#  |[] ThetaLogShell                                                 |X]|!"|
#  |"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""|"|
#  | > ThetaLog.com                                                      | |
#  | > ENV: Python 3.6.5                                                 | |
#  | > Monty Hall Problem                                                | |
#  |_____________________________________________________________________|/

# Nạp thư viện cần thiết
import random

def monty_hall_problem(trial=1000):
    """
    Mô phỏng bài toán Monty Hall
    In ra kết quả mô phỏng số lần thắng bởi lựa chọn stay và switch

    :param trial: int - số phép thử mô phỏng
    :return: tuple(stay, switch) - một tuple số lần thắng stay và switch
    """

    # Đếm số lần thắng với việc:
    # - stay: giữ nguyên lựa chọn
    # - switch: chuyển đổi lựa chọn
    stay_wins = 0
    switch_wins = 0

    for i in range(trial):
        action_door = set(range(1,4))
        # Chọn ngẫu nhiên một cửa làm xe nào
        car = random.randint(1, 3)

        # Cánh cửa mà người chơi chọn random vì không biết thông tin gì
        player = random.randint(1, 3)
        action_door.discard(player)

        open_door = None

        while not open_door:
            random_door = random.randint(1, 3)
            if random_door != car and random_door != player:
                open_door = random_door

        action_door.discard(open_door)

        switch = random.sample(action_door, 1)[0]

        if switch == car:
            switch_wins += 1
        else:
            stay_wins += 1

    print("Stay =", stay_wins)
    print("Switch = ", switch_wins)

    return stay_wins, switch_wins


if __name__ == '__main__':
    monty_hall_problem()

#   _______________________________________________________________________
#  |[] ThetaLogShell                                                 |X]|!"|
#  |"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""|"|
#  | > ThetaLog.com                                                      | |
#  | > ENV: R version 3.5.0                                              | |
#  | > Monty Hall Problem                                                | |
#  |_____________________________________________________________________|/

monty.hall.problem <- function(trial=1000){
    # Biến đếm số lần switch thắng và stay thằng
    switch.wins <- 0
    stay.wins <- 0

    # Từ khi R hạn chế vòng for nên chúng ta tạo mảng ngẫu nhiên
    car.door = sample(1:3, trial, replace=TRUE)
    player.door = sample(1:3, trial, replace=TRUE)
    
    # Mô phỏng thí nghiệm
    for (i in 1:trial) {
        open.door <- -1
        switch.door <- -1

        # Mở một cửa không có quà trong 2 cửa
        while (open.door == -1){
            door <- sample(1:3, 1, replace=TRUE)
            if (door != car.door[i] && door != player.door){
                open.door <- door
            }
        }

        # Cửa nếu lựa chọn là switch
        for (door in 1:3){
            if (door != player.door && door != open.door){
                switch.door <- door
            }
        }

        # Đếm xem stay hay switch thắng
        if (switch.door == car.door[i]){
            switch.wins <- switch.wins + 1
        }
        else {
            stay.wins <- stay.wins + 1
        }

    }

    result <- data.frame(stay.wins, switch.wins)

    # trả về kết quả
    return(result)
}
monty.hall.problem()