Сопоставить типы данных с сериями или списками

Question

Я пытаюсь отобразить dict, который содержит тип данных каждого столбца DataFrame, я пытаюсь что-то вроде этого:

data_types = {"AR3": float, 
             "AR66": float, 
             "AR10": str, 
             "AR69": str, 
             "AR61": float, 
             "AR106": float, 
             "AR109": float, 
             "AR107": str,
             "AR119": float,
             "AR1": float}


>>> df.dtypes
AR3               int64
AR55     datetime64[ns]
AR66            float64
AR10             object
AR69              int64
AR61              int64
AR106            object
AR109           float64
AR107             int64
AR119           float64
AR1              object
AR113    datetime64[ns]

######################
# What I am trying/getting:
######################

>>> df.dtypes.map(data_types)

AR3      NaN
AR55     NaN
AR66     NaN
AR10     NaN
AR69     NaN
AR61     NaN
AR106    NaN
AR109    NaN
AR107    NaN
AR119    NaN
AR1      NaN
AR113    NaN

В идеале я также хотел бы включить объекты datetime в моем отображении, но я не знал, как это сделать.

Answer 1

Вы можете попробовать это,

data_types = {
    'AR3': float,
    'AR55': 'datetime64[ns]'
    'AR66': float,
    'AR10': str,
    'AR69': str,
    'AR61': float,
    'AR106': float,
    'AR109': float,
    'AR107': str,
    'AR119': float,
    'AR1': float,
    'AR113': 'datetime64[ns]'
}

for k, v in data_types.items():
    df[k] = df[k].astype(v)