Probabilidade em Python

Por

Código em Python que calcula probabilidades: define uma função para determinar a probabilidade de um evento ocorrer e a probabilidade de não ocorrer, usando conceitos básicos de probabilidade.

🟢 Probabilidade

# Probabilidade

import numpy as np
import pandas as pd

enem_sp = pd.read_csv('/content/drive/MyDrive/Colab Notebooks/Seção 3/enem_2019_tratado.csv', sep=',', encoding='iso-8859-1')

enem_sp.head()
# Saída:
#    NU_INSCRICAO  NU_ANO  CO_MUNICIPIO_RESIDENCIA  ...  NU_NOTA_MT  TP_PRESENCA_LC  NU_NOTA_LC
# 0    2000000001    2019                  3550308  ...       593.2               1       530.5
# 1    2000000002    2019                  3550308  ...       689.7               1       610.2
# ...

enem_Guarulhos = enem_sp.loc[enem_sp.NO_MUNICIPIO_RESIDENCIA == 'Guarulhos']

enem_Guarulhos.head()
# Saída:
#    NU_INSCRICAO  NO_MUNICIPIO_RESIDENCIA  ...  NU_NOTA_MT  NU_NOTA_LC
# 0    2000000006           Guarulhos       ...     612.3       585.1
# ...

enem_Bauru = enem_sp.loc[enem_sp.NO_MUNICIPIO_RESIDENCIA == 'Bauru']

enem_Bauru.head()
# Saída:
#    NU_INSCRICAO  NO_MUNICIPIO_RESIDENCIA  ...  NU_NOTA_MT  NU_NOTA_LC
# 0    2000000012           Bauru           ...     577.0       521.7
# ...

len(enem_Guarulhos)
len(enem_Bauru)
len(enem_sp)

# Saída:
# len(enem_Guarulhos) -> 2150
# len(enem_Bauru)     -> 1720
# len(enem_sp)        -> 50000

🟢 Função de Probabilidade:

# Função de probabilidade

# Definindo uma função
def probab(A, E):
    resultado = (A / E) * 100
    print('{:.2f}'.format(resultado))

# Probabilidade de retirar um vestibulando de Guarulhos
probab(len(enem_Guarulhos), len(enem_sp))

# Probabilidade de retirar um vestibulando de Bauru
probab(len(enem_Bauru), len(enem_sp))

# Saída esperada:
# 4.30
# 3.44

🟢Probabilidade de não ocorrer um evento:

# Probabilidade de **não** ocorrer um evento

def probab_nao(A, E):
    resultado = (1 - (A / E)) * 100
    print('{:.2f}'.format(resultado))

probab_nao(len(enem_Guarulhos), len(enem_sp))
probab_nao(len(enem_Bauru), len(enem_sp))

# Probabilidade de não retirar um vestibulando de Guarulhos nem de Bauru
probab_nao(len(enem_Guarulhos) + len(enem_Bauru), len(enem_sp))

# Saída esperada:
# 95.70
# 96.56
# 92.26

Deixe um comentário