Summary  
This chapter explains how to transform tabular data into a pivot table by grouping records on one or more fields and applying aggregation functions to compute summary metrics. It covers the concept of multi-dimensional data summarization and restructuring.

General domain of usage  
Data analysis

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(df1, df2):
    """ÐÐµÑÐµÐ²ÑÑÑÑ ÑÑÐ²Ð½ÑÑÑÑ Ð´Ð²Ð¾Ñ DataFrame Ð½ÐµÐ·Ð°Ð»ÐµÐ¶Ð½Ð¾ Ð²ÑÐ´ ÑÐ¾ÑÑÑÐ²Ð°Ð½Ð½Ñ ÑÐ½Ð´ÐµÐºÑÑÐ²."""
    try:
        df1_sorted = df1.sort_index()
        df2_sorted = df2.sort_index()
        return df1_sorted.equals(df2_sorted)
    except Exception:
        return False


class TestPivotTable(unittest.TestCase):
    def test_pivot_table_result(self):
        """
        Check if pivot table correctly groups by Airline and AirportFrom,
        counts Delay values, and finds min and max of Length.
        """
        import user_code

        # Reference dataset
        url = "https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/plane"
        df = pd.read_csv(url, index_col=0)

        # Expected result
        ref = pd.pivot_table(
            df,
            index=["Airline", "AirportFrom"],
            values=["Delay", "Length"],
            aggfunc={"Length": ["min", "max"], "Delay": "count"},
        )

        # Try to find result variable
        user_result = None
        if hasattr(user_code, "data"):
            candidate = getattr(user_code, "data")
            if isinstance(candidate, pd.DataFrame):
                user_result = candidate

        assert user_result is not None, "The resulting pivot table (variable 'data') is missing or invalid."

        # Compare DataFrames
        condition = frames_equal(user_result, ref)
        _dynamic_test(
            self,
            condition,
            "The pivot table correctly calculates min/max Length and counts Delay values per Airline and AirportFrom.",
            "The pivot table result is incorrect. Check the index, values, or aggfunc arguments."
        )


class TestOutput(unittest.TestCase):
    def test_output_head_10(self):
        """
        Check that the output of print(data.head(10)) is not empty.
        """
        import user_code
        captured_output = io.StringIO()
        sys.stdout = captured_output
        print(user_code.data.head(10))
        sys.stdout = sys.__stdout__

        output_text = captured_output.getvalue().strip()
        condition = len(output_text) > 0
        _dynamic_test(
            self,
            condition,
            "The first 10 rows of the pivot table are printed correctly using .head(10).",
            "The output is missing or incorrect. Use print(data.head(10))."
        )


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

test_code.py

Este curso contiene muchas funciones útiles para un futuro analista de datos. Aprenderá diferentes formas de extraer datos e incluso establecer condiciones sobre ellos. Después de esto, estará familiarizado con los métodos de agrupación de datos. Además, aprenderá cómo preprocesar datos. Cada sección tiene su propio conjunto de datos, lo que hace que el curso sea interesante.

Esta sección le enseñará cómo mostrar columnas específicas por sus títulos o índices. Además, se familiarizará con las formas en que puede seleccionar filas por sus índices.

Aquí aprenderá a extraer datos que cumplen condiciones específicas. Además, aprenderá a combinarlas e incluso a crear las suyas propias.

En esta sección, ampliará su conocimiento sobre el establecimiento de diferentes condiciones de datos. Aprenderá a verificar si sus datos se encuentran en una lista definida de valores o entre dos valores. También aprenderá a encontrar los valores más grandes y más pequeños.

Esta sección es una de las más fascinantes del curso. Aquí aprenderá cómo agrupar datos de diferentes maneras. Le ayudará a trabajar como analista de datos para obtener información sobre grupos de datos específicos.

Esta sección es una de las más significativas para un analista de datos, ya que si los datos contienen valores faltantes o en un formato incorrecto, será imposible trabajar con ellos. Por lo tanto, aquí aprenderá cómo tratar dichos valores inapropiados.

Creación de una Tabla Dinámica

Solución

Creación de una Tabla Dinámica

Solución