Crear un DataFrame a partir de una Lista

Escribe una solución para crear un DataFrame a partir de una lista 2D llamada student_data. Esta lista contiene los IDs y las edades de algunos estudiantes. El DataFrame debe tener dos columnas, student_id y age, y mantener el mismo orden que la lista 2D original.

El formato del resultado se muestra en el siguiente ejemplo.

Ejemplo 1:

Entrada:

student_data = [
    [1, 15],
    [2, 11],
    [3, 11],
    [4, 20]
]

Salida:

+------------+-----+
| student_id | age |
+------------+-----+
| 1          | 15  |
| 2          | 11  |
| 3          | 11  |
| 4          | 20  |
+------------+-----+

Explicación:
- Un DataFrame fue creado a partir de student_data, con dos columnas student_id y age.

Solución

import pandas as pd

from typing import List


def createDataframe(student_data: List[List[int]]) -> pd.DataFrame:
    return pd.DataFrame(student_data, columns=['student_id', 'age'])


def test_createDataframe():
    column_names = ['student_id', 'age']
    data = [[1, 15], [2, 11], [3, 11], [4, 20]]

    exp = pd.DataFrame(data, columns=column_names)
    got = createDataframe(data)

    pd.testing.assert_frame_equal(got, exp)