INNER CODE UNIT · Python

star_wars_data

aeturrell/coding-for-economists · data_set_prep.py:13

def star_wars_data():
    """Saves star wars character data with set
    datatypes and in pickle format.
    """
    df = pd.read_csv(
        os.path.join("data", "characters.csv"),
        thousands=",",
        dtype={
            "name": "string",
            "height": float,
            "mass": float,
            "hair_color": "category",
            "skin_color": "category",
            "eye_color": "category",
            "birth_year": "string",
            "gender": "category",
            "homeworld": "category",
            "species": "category",

View source record →