Pandas创建指定Schema空DataFrame报错：不支持无单位datetime64类型

问题原因 #

新版本的Numpy和Pandas对datetime64类型的校验更严格，不再支持无单位的np.datetime64类型，必须明确指定时间单位（如datetime64[ns]），以此避免类型定义的模糊性。

修正方案 #

以下两种方案任选其一即可：

方式1：修改Schema中的时间类型为带单位的格式

直接将np.datetime64替换为带单位的np.dtype('datetime64[ns]')，或者更简洁的字符串格式'datetime64[ns]'（Pandas原生支持该字符串类型标识）：

import pandas as pd
import numpy as np

# 方案1a：使用带单位的numpy dtype
schema = {'timestamp': np.dtype('datetime64[ns]'), 
          'instrument_token': int, 
          'last_price': float, 
          'volume': int}
data = pd.DataFrame(columns=schema.keys()).astype(schema)

# 方案1b：使用字符串类型标识（更简洁）
schema = {'timestamp': 'datetime64[ns]', 
          'instrument_token': int, 
          'last_price': float, 
          'volume': int}
data = pd.DataFrame(columns=schema.keys()).astype(schema)

方式2：直接创建指定类型的空Series组成DataFrame

这种方式无需先创建空DataFrame再做类型转换，效率更高：

import pandas as pd
import numpy as np

schema = {'timestamp': 'datetime64[ns]', 
          'instrument_token': int, 
          'last_price': float, 
          'volume': int}
# 为每个列创建对应类型的空Series，再组合成DataFrame
data = pd.DataFrame({col: pd.Series(dtype=dtype) for col, dtype in schema.items()})

验证 #

执行修正后的代码，可得到类型完全匹配Schema的空DataFrame：

print(data.dtypes)
# 输出示例：
# timestamp          datetime64[ns]
# instrument_token             int64
# last_price                 float64
# volume                       int64
# dtype: object

内容的提问来源于stack exchange，提问作者lenmet