# Exercício Prático 2 - Cálculo Simplificado de TF-IDF
import math

doc1 = ["big", "data", "python"]
doc2 = ["python", "codigo", "dados"]
corpus = [doc1, doc2]

palavra_alvo = "big"

# 1. Cálculo do TF no Documento 1
tf_doc1 = doc1.count(palavra_alvo) / len(doc1)

# 2. Cálculo do IDF no Corpus
docs_com_palavra = sum(1 for doc in corpus if palavra_alvo in doc)
idf = math.log(len(corpus) / docs_com_palavra) if docs_com_palavra > 0 else 0

# 3. TF-IDF
tfidf = tf_doc1 * idf

print("=== CÁLCULO DE TF-IDF ===")
print(f"Palavra Analisada: '{palavra_alvo}'")
print(f"TF no Doc 1: {tf_doc1:.4f}")
print(f"IDF no Corpus: {idf:.4f}")
print(f"TF-IDF Resultante: {tfidf:.4f}")