Olá , Thiago! tudo traquilo ?
Parabéns por ter pensado em outra solução que não a demonstrada pelo instrutor, explorar outras possibilidades é essencial para o aprendizado.
Só um pequeno detalhe na solução apresentada por você, antes de utilizar o método count faz-se necessário o casting, que é transformar um tipo de objeto em outro. Quando o professor executa o código abaixo, nos é retornado um objeto do tipo numpy.ndarray, que não possui o método count.
X = Xdummies_df.values
Y = Ydummies_df.values
Para verificar o tipo do objeto, utiliza-se o método type(), conforme exemplo abaixo:
print(type(Y))
Saída:
<class 'numpy.ndarray'>
Logo para a utilização do método count, faz-se necessário primeiro transformar de numpy.ndarray para lista, conforme exemplo abaixo (foi utilizado a mesma variável Y criada pelo instrutor):
list(Y).count(1)
Saída:
832
O código acima nos retorna quantos 1’s o objeto Y, transformado em uma lista, possui. Essa mesma ideia é aplicada as outras variáveis criadas pelo instrutor (treino_dados, treino_marcacoes...).
Caso queira se aprofundar um pouco mais sobre numpy.ndarray recomendo a leitura nesse link.
Caso ainda persista alguma dúvida, estou sempre à disposição.
Espero ter ajudado.
:)
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓.Bons Estudos!