Como adicionar Coloumn com base em outra coluna

metadata['class_pnemonia'] = metadata['class'].astype('category').map({0: 'HEALTHY ', 1: 'PNEUMONIA'})
Lovely Llama