python 中的函数什么都不返回

Question

我有这个数组，

     a = array([1,5,7])

我应用了 where 函数

     where(a==8)

在这种情况下返回的是

    (array([], dtype=int64),)

但是我希望代码在 where 函数返回空数组时返回整数“0”。这可能吗？

Answer 1

def where0(vec):
    a = where(vec)
    return a if a[0] else 0
    # The return above is equivalent to:
    # if len(a[0]) == 0:
    #     return 0  # or whatever you like
    # else:
    #     return a

a = array([1,5,7])
print where0(a==8)

还要考虑您的问题下 aix 的评论。不要修复 where()，而是修复你的算法

Answer 2

最好使用只有一种返回类型的函数。您可以检查数组的大小以了解它是否为空，这应该可以完成工作：

 a = array([1,5,7])
 result = where(a==8)

 if result[0] != 0:
     doFancyStuff(result)
 else:
     print "bump"

Answer 3

空数组将返回 0 和 .size

import numpy as np    
a = np.array([])    
a.size
>> 0

Answer 4

尝试下面的方法。这将处理返回索引 0 时等于 0 的测试失败的情况。 （例如以下情况下的 np.where(a==1)）

a = array([1,5,7])
ret = np.where(a==8)
ret = ret if ret[0].size else 0