使用pandas.convert_dtype指定pyarrow后端报错：ArrowDtype未定义

1. 确认PyArrow依赖已安装 #

pandas 2.1.0使用PyArrow作为 dtype 后端需要依赖PyArrow库，先检查环境是否已安装：

pip list | grep pyarrow

若未安装，执行以下命令安装适配版本（pandas 2.1.0要求PyArrow≥15.0.0）：

pip install pyarrow>=15.0.0

2. 显式导入ArrowDtype类 #

在代码开头添加导入语句，确保pandas能正确识别该类型：

from pandas.core.arrays.arrow import ArrowDtype

3. 修正代码中的兼容问题 #

原代码里np.random.random_integers已被弃用，替换为np.random.randint，同时补充缺失的模块导入：

import numpy as np
import pandas as pd
import datetime
from pandas.core.arrays.arrow import ArrowDtype

df = pd.DataFrame({'A' : ['spam', 'eggs', 'spam', 'eggs'] * 2,
                   'B' : ['alpha', 'beta', 'gamma' , 'foo'] * 2,
                   'C' : [np.random.choice(pd.date_range(datetime.datetime(2013,1,1),datetime.datetime(2013,1,3))) for i in range(8)],
                   'D' : np.random.randn(8),
                   'E' : np.random.randint(1, 9, size=8)}).convert_dtypes(dtype_backend='pyarrow')

4. 验证转换结果 #

运行代码后，通过以下命令查看列类型，确认已切换为PyArrow后端：

print(df.dtypes)

预期输出示例：

A    string[pyarrow]
B    string[pyarrow]
C    datetime64[ns][pyarrow]
D    float64[pyarrow]
E      int64[pyarrow]
dtype: object

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