如何判断给定dtype是否为多个numpy dtype的子类型?
解法1:兼容所有numpy版本的通用写法
用Python内置的any()函数配合生成器表达式,把需要匹配的父类型统一放在元组里遍历校验,逻辑和原生类型的in判断完全对应:
import numpy as np # 定义要匹配的目标父类型集合,这里用np.bool_、np.floating避免弃用警告 allowed_types = (np.bool_, np.integer, np.floating) my_var = 5 # 一行完成校验,无需多个or拼接 result = any(np.issubdtype(type(my_var), t) for t in allowed_types) print(result) # 输出 True
解法2:numpy >= 1.21.0 专属极简写法
新版本的numpy.issubdtype已经支持第二个参数直接传入类型元组,只要匹配其中任意一个父类型就会返回True,写法和原生类型校验几乎一致:
result = np.issubdtype(type(my_var), (np.bool_, np.integer, np.floating)) print(result) # 输出 True
补充说明
- 若校验对象本身就是numpy dtype实例(而非变量),不需要包裹
type(),直接传入即可,例如np.issubdtype(np.int64, allowed_types)返回True - 已废弃的
np.bool、np.float别名建议替换为np.bool_、np.floating,避免运行时警告
内容的提问来源于stack exchange,提问作者Hawklaz
相关产品推荐
相关产品推荐

