Solucionado (ver solução)
Solucionado
(ver solução)
1
resposta

Como selecionar linhas com base no dado de uma coluna?

Quero selecionar e mostrar os dados das duas primeiras colunas das linhas em que a pessoa tem sexo masculino, porém, pensando que tenho milhares de linhas de código, como fazer essa validação?

dados = np.array(
    [
        ['Roberto', 'casado', 'masculino'],
        ['Sheila', 'solteiro', 'feminino'],
        ['Bruno', 'solteiro', 'masculino'],
        ['Rita', 'casado', 'feminino']
    ]
)

dados[dados[-1] == 'masculino', :2]

Por que não posso fazer a validação no campo de seleção das linhas que desejo? (As que possuem masculino na ultima coluna). Qual seria uma alternativa correta para o código?

1 resposta
solução!

Oi Nawan,

Falta só um detalhe para fazer a seleção corretamente:

import numpy as np

dados = np.array(
    [
        ['Roberto', 'casado', 'masculino'],
        ['Sheila', 'solteiro', 'feminino'],
        ['Bruno', 'solteiro', 'masculino'],
        ['Rita', 'casado', 'feminino']
    ]
)

dados[dados[:, -1] == 'masculino', :2]

Espero ter ajudado

Quer mergulhar em tecnologia e aprendizagem?

Receba a newsletter que o nosso CEO escreve pessoalmente, com insights do mercado de trabalho, ciência e desenvolvimento de software