Python 中变量的数组或标量类型如何区分？

2024-03-24 23:22:15

如何识别 Python 中变量的数组或标量类型

问题陈述

在 Python 中处理数据时，区分变量是数组还是标量至关重要。数组包含多个元素，而标量仅表示单个值。在某些情况下，确定变量的类型可能很棘手，尤其是当你处理复杂的自定义对象时。

解决方法

有多种方法可以检查 Python 中变量的数组或标量类型：

isinstance() 函数

import numpy as np

def is_array(variable):
    return isinstance(variable, np.ndarray)

def is_scalar(variable):
    return isinstance(variable, (int, float, complex))

type() 函数

def is_array(variable):
    return type(variable) == np.ndarray

def is_scalar(variable):
    return type(variable) in (int, float, complex)

hasattr() 函数

def is_array(variable):
    return hasattr(variable, "__len__")

def is_scalar(variable):
    return not hasattr(variable, "__len__")

len() 函数

def is_array(variable):
    try:
        len(variable)
        return True
    except TypeError:
        return False

def is_scalar(variable):
    try:
        len(variable)
        return False
    except TypeError:
        return True

示例用法

import numpy as np

# 标量
scalar = 50

# 数组
array = np.array([0, 10, 20, 30])

# 使用 isinstance() 函数
print(isinstance(scalar, np.ndarray))  # False
print(isinstance(array, np.ndarray))  # True

# 使用 type() 函数
print(type(scalar) == np.ndarray)  # False
print(type(array) == np.ndarray)  # True

# 使用 hasattr() 函数
print(hasattr(scalar, "__len__"))  # False
print(hasattr(array, "__len__"))  # True

# 使用 len() 函数
try:
    len(scalar)
    print("Scalar has length")
except TypeError:
    print("Scalar has no length")

try:
    len(array)
    print("Array has length")
except TypeError:
    print("Array has no length")