Quais ações se movem juntas?
No exercício anterior, você agrupou as empresas de acordo com as variações diárias dos preços das ações. Então, quais empresas têm preços de ações que tendem a mudar da mesma maneira? Agora você vai dar uma olhada nos rótulos dos clusters do seu agrupamento pra descobrir.
A sua solução para o exercício anterior já foi executada. Lembre-se de que você criou um Pipeline pipeline
com um modelo KMeans
e ajustou ele à matriz NumPy movements
dos movimentos diários das ações. Além disso, tem uma lista companies
com os nomes das empresas.
Este exercício faz parte do curso
Aprendizado não supervisionado em Python
Instruções do exercício
- Importe
pandas
comopd
. - Use o método “
.predict()
” do pipeline para prever os rótulos para “movements
”. - Alinhe os rótulos do cluster com a lista de nomes de empresas
companies
criando um DataFramedf
comlabels
ecompanies
como colunas. Isso foi feito para você. - Use o método “
.sort_values()
” de “df
” para classificar o DataFrame pela coluna “'labels'
” e imprima o resultado. - Clique em enviar e veja quais empresas estão juntas em cada grupo!
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Import pandas
import pandas as pd
# Predict the cluster labels: labels
labels = ____
# Create a DataFrame aligning labels and companies: df
df = pd.DataFrame({'labels': labels, 'companies': companies})
# Display df sorted by cluster label
print(____)