欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

TypeError: unsupported operand type(s) for /: 'NoneType' and 'NoneType'

程序员文章站 2022-03-05 12:20:41
...

当你使用spark udf时,进行如下操作

@udf(ArrayType(DoubleType()))
def listdivision(alist,blist):
    if (np.array(blist).sum() is None)|(np.array(alist).sum() is None):
        return None
    else:
        result=[]
        label = np.array(blist)==0
        result = np.array(alist)/np.array(blist)
        result[label] = 1
        return result

其核心就是np.array的除法,当array内部存在array([None])时就会造成这样的报错

TypeError: unsupported operand type(s) for /: 'NoneType' and 'NoneType'

所以需要使用np.array(blist).sum() is None进行判断