如何检查类数组对象及ndarray向目标dtype的安全可转换性？

1. 检查类数组序列（列表/元组/ndarray）能否安全转换为目标dtype #

要兼容列表、元组这类类数组对象，核心是先将其转换为numpy数组，再复用基于实际值的检查逻辑。可以实现一个通用函数，先把输入转为ndarray，再调用针对ndarray的检查函数：

import numpy as np

def is_safe_cast_array_like(arr_like, target_dtype):
    target_dtype = np.dtype(target_dtype)
    # 将类数组转为ndarray，保留原始数据信息
    arr = np.asarray(arr_like)
    return is_safe_cast_ndarray(arr, target_dtype)

2. 基于ndarray实际值检查安全转换 #

np.can_cast仅依据类型判断，不考虑实际数据范围，我们需要结合目标dtype的数值范围和数据本身的特性来判断：

步骤1：获取目标dtype的数值上下限 #

先写一个工具函数，根据dtype类型（整数/浮点数）获取对应的范围：

def get_dtype_bounds(dtype):
    dtype = np.dtype(dtype)
    if dtype.kind == 'f':
        info = np.finfo(dtype)
        return info.min, info.max
    elif dtype.kind in ('i', 'u'):
        info = np.iinfo(dtype)
        return info.min, info.max
    # 处理bool等特殊类型
    return dtype.type(0), dtype.type(1)

步骤2：实现基于实际值的检查逻辑 #

针对不同的源类型和目标类型组合做针对性判断：

def is_safe_cast_ndarray(arr, target_dtype):
    target_dtype = np.dtype(target_dtype)
    arr_dtype = arr.dtype

    # 特殊情况：目标是整数，源是浮点数，必须所有元素都是整数
    if target_dtype.kind in ('i', 'u') and arr_dtype.kind == 'f':
        # 检查是否存在非整数元素
        if not np.all(np.mod(arr, 1) == 0):
            return False
        # 转为整数后再检查范围
        arr_int = arr.astype(np.int64)
        min_bound, max_bound = get_dtype_bounds(target_dtype)
        return arr_int.min() >= min_bound and arr_int.max() <= max_bound
    
    # 特殊情况：目标是无符号整数，源不能有负数
    if target_dtype.kind == 'u' and arr.min() < 0:
        return False
    
    # 通用范围检查
    min_bound, max_bound = get_dtype_bounds(target_dtype)
    arr_min = arr.min()
    arr_max = arr.max()
    
    return arr_min >= min_bound and arr_max <= max_bound

测试示例 #

• is_safe_cast_array_like([1, 2, 3], 'int32') → True
• is_safe_cast_array_like(np.array([1], dtype='int64'), 'int32') → True
• is_safe_cast_array_like([1.1, 2.2], 'int32') → False（存在非整数浮点数）
• is_safe_cast_array_like([-1], 'uint32') → False（无符号整数不能存负数）
• is_safe_cast_array_like([2**30], 'int32') → True（2^30在int32范围内）
• is_safe_cast_array_like([2**31], 'int32') → False（2^31超出int32上限）

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