引言

不知道大家有没有过和我一样的困惑，就是不知道该如何判断一个数据的类型。比如整数类型以及 ndarray 类型。这里介绍两种方法来实现这一判断。

正文

对于一个整数型数据

# 整数a
a = 1
if isinstance(a, int):
    print(a)

if type(a) == int:
    print(a)

if type(a) is int:
    print(a)
"""
result:
1
1
1
"""

可以看到上述三种方式均成功输出了结果，上述方法对其他数据类型也适用。

ndarray类型判断

import numpy as np

a = np.array(1)
if type(a) == np.ndarray:
    print(type(a))

if isinstance(a, np.ndarray):
    print(a)

if type(a) is np.ndarray:
    print(a)
"""
result:
<class 'numpy.ndarray'>
1
1
"""

可以看到，上述三种方式也都获取到了结果。

虽然对于数据类型判断，上述三种方法均适用，但是为了避免一些不必要的麻烦和获取到更加可信的结果，推荐使用 isinstance() 函数进行判断。

拓展示例

如果我们想要判断输入的数据是否为 int 或者 float 类型中的一种该怎么办呢？

a = 1
b = 1.0

if isinstance(a, (int, float)):
    print(a)

if isinstance(b, (int, float)):
    print(b)
"""
result:
1
1.0
"""

可以看到，这里我们也得到了想要的结果。

需要特别注意的是，多个数据类型进行判断时，需要使用元组的写法，如 (int, float)，这里如果使用数组的写法会报错。

如果大家觉得有用，就请点个赞吧~