python:如何识别变量是数组还是标量

Question

我有一个接受参数的函数NBins.我想用标量50或数组调用这个函数[0, 10, 20, 30].如何识别功能,长度NBins是多少？或者换句话说,如果它是标量或向量？

我试过这个:

>>> N=[2,3,5]
>>> P = 5
>>> len(N)
3
>>> len(P)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: object of type 'int' has no len()
>>> 

正如你看到的,我不能申请len到P,因为它不是一个数组....有什么样isarray或isscalar在Python？

谢谢

Answer 1

>>> isinstance([0, 10, 20, 30], list)
True
>>> isinstance(50, list)
False

要支持任何类型的序列,请检查collections.Sequence而不是list.

注意:isinstance还支持一个类的元组,type(x) in (..., ...)应该避免检查并且是不必要的.

你可能还想检查一下 not isinstance(x, (str, unicode))

Answer 2

以前的答案假设该数组是python标准列表.作为经常使用numpy的人,我推荐一个非常pythonic测试:

if hasattr(N, "__len__")

Answer 3

将@jamylak和@jpaddison3的答案结合在一起,如果你需要对numpy数组作为输入强大并以与列表相同的方式处理它们,你应该使用

import numpy as np
isinstance(P, (list, tuple, np.ndarray))

这对list,tuple和numpy数组的子类很有用.

如果你想要对所有其他序列子类(不仅仅是列表和元组)具有健壮性,请使用

import collections
import numpy as np
isinstance(P, (collections.Sequence, np.ndarray))

为什么要用这种方式做事isinstance而不是type(P)与目标价值进行比较？这是一个例子,我们制作和研究NewList列表的一个简单子类的行为.

>>> class NewList(list):
...     isThisAList = '???'
... 
>>> x = NewList([0,1])
>>> y = list([0,1])
>>> print x
[0, 1]
>>> print y
[0, 1]
>>> x==y
True
>>> type(x)
<class '__main__.NewList'>
>>> type(x) is list
False
>>> type(y) is list
True
>>> type(x).__name__
'NewList'
>>> isinstance(x, list)
True

尽管x和y比较相同,处理它们type会导致不同的行为.然而,由于x是的子类的实例list,使用isinstance(x,list)得到所需的行为和治疗x和y以相同的方式.

Answer 4

numpy中是否有等同于isscalar()的东西？是.

>>> np.isscalar(3.1)
True
>>> np.isscalar([3.1])
False
>>> np.isscalar(False)
True

Answer 5

虽然@ jamylak的方法更好,但这是一种替代方法

>>> N=[2,3,5]
>>> P = 5
>>> type(P) in (tuple, list)
False
>>> type(N) in (tuple, list)
True

Answer 6

另一种替代方法（使用类名属性）：

N = [2,3,5]
P = 5

type(N).__name__ == 'list'
True

type(P).__name__ == 'int'
True

type(N).__name__ in ('list', 'tuple')
True

无需导入任何东西。

Answer 7

这是我发现的最好方法：检查__len__和的存在__getitem__。

你可能会问为什么？原因包括：

1. 流行的方法isinstance(obj, abc.Sequence)在包括 PyTorch 的 Tensor 在内的某些对象上失败，因为它们没有实现__contains__.
2. 不幸的是，Python 的 collections.abc 中没有任何内容可以检查 only __len__，__getitem__我认为这是类似数组对象的最小方法。
3. 它适用于列表、元组、ndarray、张量等。

言归正传：

def is_array_like(obj, string_is_array=False, tuple_is_array=True):
    result = hasattr(obj, "__len__") and hasattr(obj, '__getitem__') 
    if result and not string_is_array and isinstance(obj, (str, abc.ByteString)):
        result = False
    if result and not tuple_is_array and isinstance(obj, tuple):
        result = False
    return result

请注意，我添加了默认参数，因为大多数时候您可能希望将字符串视为值，而不是数组。对于元组也是如此。