Summary  
Demonstrates how to use boolean expressions to filter and select subsets of data from collections based on conditional criteria.

General domain of usage  
Data analysis

Para se destacar em programação, é sempre necessário praticar. Portanto, nesta tarefa, também é preciso memorizar as funções do capítulo anterior. Se você ainda não domina as funções do último capítulo, consulte as dicas para revisá-las.

import unittest
import pandas as pd
import io
import sys


def _dynamic_test(test_case, condition, success_msg, failure_msg):
    if condition:
        test_case._testMethodName = success_msg
        test_case.assertTrue(True, success_msg)
    else:
        test_case._testMethodName = failure_msg
        test_case.fail(failure_msg)


def frames_equal_ignore_col_order(df1, df2):
    """ÐÐµÑÐµÐ²ÑÑÑÑ ÑÑÐ²Ð½ÑÑÑÑ DataFrame Ð½ÐµÐ·Ð°Ð»ÐµÐ¶Ð½Ð¾ Ð²ÑÐ´ Ð¿Ð¾ÑÑÐ´ÐºÑ ÐºÐ¾Ð»Ð¾Ð½Ð¾Ðº."""
    if set(df1.columns) != set(df2.columns):
        return False
    common_cols = sorted(df1.columns)
    df1_sorted = df1[common_cols].reset_index(drop=True)
    df2_sorted = df2[common_cols].reset_index(drop=True)
    return df1_sorted.equals(df2_sorted)


class TestExtraction(unittest.TestCase):
    def test_extract_columns(self):
        """
        1. Extract all rows from columns 'name' and 'hazardous' using .loc[].
        """
        import user_code
        df = pd.read_csv(
            "https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/planet"
        )
        ref = df.loc[:, ["name", "hazardous"]]

        condition = (
            hasattr(user_code, "data_extracted")
            and isinstance(user_code.data_extracted, pd.DataFrame)
            and frames_equal_ignore_col_order(user_code.data_extracted, ref)
        )
        _dynamic_test(
            self,
            condition,
            "Columns 'name' and 'hazardous' are extracted correctly using .loc[].",
            "The extraction of 'name' and 'hazardous' columns is incorrect."
        )


class TestFiltering(unittest.TestCase):
    def test_filter_true_rows(self):
        """
        2. Filter rows where 'hazardous' is True.
        """
        import user_code
        ref = user_code.data_extracted.loc[user_code.data_extracted["hazardous"] == True]

        condition = (
            hasattr(user_code, "data_filtered")
            and isinstance(user_code.data_filtered, pd.DataFrame)
            and frames_equal_ignore_col_order(user_code.data_filtered, ref)
        )
        _dynamic_test(
            self,
            condition,
            "Rows where 'hazardous' == True are filtered correctly.",
            "Filtering by 'hazardous' == True is incorrect."
        )


class TestOutput(unittest.TestCase):
    def test_print_output(self):
        """
        3. Output the first five rows of data_filtered using .head().
        """
        import user_code

        captured_output = io.StringIO()
        sys.stdout = captured_output
        print(user_code.data_filtered.head())
        sys.stdout = sys.__stdout__

        output_text = captured_output.getvalue().strip()
        condition = len(output_text) > 0
        _dynamic_test(
            self,
            condition,
            "The output displays the first five rows of 'data_filtered' correctly.",
            "The 'data_filtered' dataset is not printed correctly. Use print(data_filtered.head())."
        )


if __name__ == "__main__":
    unittest.main()


test_code.py

Este curso contém diversas funções úteis para um futuro analista de dados. Você aprenderá diferentes formas de extrair dados e até mesmo definir condições sobre eles. Após isso, estará familiarizado com os métodos de agrupamento de dados. Além disso, você aprenderá como pré-processar dados. Cada seção possui seu próprio conjunto de dados, tornando o curso envolvente.

Esta seção ensinará como exibir colunas específicas pelos seus títulos ou índices. Além disso, você se familiarizará com as formas de selecionar linhas por índices.

Aqui, você aprenderá como extrair dados que possuem condições específicas. Além disso, você aprenderá como combiná-las e até mesmo criar as suas próprias.

Nesta seção, você expandirá seu conhecimento sobre como definir diferentes condições de dados. Você aprenderá a verificar se seus dados estão em uma lista definida de valores ou entre dois valores. Você também aprenderá a encontrar os maiores e menores valores.

Esta seção é uma das mais fascinantes do curso. Aqui, você aprenderá como agrupar dados de diferentes maneiras. Isso ajudará no trabalho como analista de dados para obter informações sobre grupos de dados específicos.

Esta seção é uma das mais significativas para um analista de dados, pois, se os dados contiverem valores ausentes ou em formato incorreto, será impossível trabalhar com eles. Assim, você aprenderá como lidar com esses valores inadequados aqui.

Selecionando Dados com Base em Condição

Solução