seleção aleatória de pacientes

[@revisaosistematica6485]

import random # Define códigos de cores ANSI RED = '\033[91m' YELLOW = '\033[93m' BLUE = '\033[94m' GREEN = '\033[92m' WHITE = '\033[0m' # Função para imprimir uma linha de pontos def print_line(): print("·" * 70) # Função para imprimir uma linha com o texto centralizado def print_centered(text): spaces = (70 - len(text)) // 2 print("·" + " " * spaces + text + " " * spaces + "·") # Função para imprimir os dados inseridos em uma tabela de pontos def print_input_data(total_pacientes, pacientes_selecionados, limites_meses): print_line() print_centered("Dados inseridos:") print_line() print(f"· Número total de pacientes: {RED}{total_pacientes}{WHITE}") print(f"· Número de pacientes a serem selecionados: {YELLOW}{pacientes_selecionados}{WHITE}") print_line() for mes, limites in limites_meses.items(): print(f"· Limite superior dos pacientes admitidos em {GREEN}{mes}{WHITE}: {limites[0]} - {limites[1]}") print_line() # Função para imprimir os resultados em uma tabela de pontos def print_results(pacientes_por_mes): print_line() print_centered("Resultados:") print_line() for mes, pacientes in pacientes_por_mes.items(): total_pacientes_mes = len(pacientes) porcentagem_mes = (total_pacientes_mes / pacientes_selecionados) * 100 pacientes_selecionados_str = ', '.join(map(str, sorted(pacientes))) print(f"{GREEN}· Pacientes selecionados em {mes}:{WHITE} {pacientes_selecionados_str}") print(f"{RED}· Total de pacientes em {mes}: {total_pacientes_mes} ({porcentagem_mes:.2f}%)") print_line() # Pergunta ao usuário o número total de pacientes total_pacientes = int(input(f"· Digite o número total de pacientes: {RED}")) # Pergunta ao usuário o número de pacientes a serem selecionados pacientes_selecionados = int(input(f"· Digite o número de pacientes a serem selecionados: {YELLOW}")) # Cria um dicionário para armazenar os limites dos pacientes por mês limites_meses = {} # Preenche o número de pacientes admitidos em cada mês meses = ["Janeiro", "Fevereiro", "Março", "Abril", "Maio", "Junho", "Julho", "Agosto", "Setembro", "Outubro", "Novembro", "Dezembro"] for mes in meses: limites_str = input(f"· Digite o limite superior dos pacientes admitidos em {GREEN}{mes}{WHITE} (formato: inicio-fim ou inicio fim): ") limites = [int(num) for num in limites_str.replace('-', ' ').split()] limite_superior = max(limites) limites_meses[mes] = (min(limites), limite_superior) # Calcula a proporção de pacientes a serem selecionados em cada mês proporcao_meses = {} for mes, limites in limites_meses.items(): proporcao = (limites[1] - limites[0] + 1) / total_pacientes proporcao_meses[mes] = round(proporcao * pacientes_selecionados) # Seleciona aleatoriamente os números correspondentes a cada paciente por mês pacientes_por_mes = {} for mes, limites in limites_meses.items(): inicio, fim = limites proporcao = proporcao_meses[mes] if proporcao == 0: pacientes_mes = [] elif proporcao >= fim - inicio + 1: pacientes_mes = list(range(inicio, fim + 1)) else: pacientes_mes = random.sample(range(inicio, fim + 1), proporcao) pacientes_por_mes[mes] = pacientes_mes # Imprime os dados inseridos e os resultados print_line() print_input_data(total_pacientes, pacientes_selecionados, limites_meses) print_results(pacientes_por_mes)