你的位置:首页 > 信息动态 > 新闻中心
信息动态
联系我们

使用numpy遇到ValueError: The truth value of an array with more than one element is ambiguous

2021/12/31 0:39:58

Python中使用numpy和pandas时,遇到如标题所示的报错,完整报错是:

写了一个函数,参数均为ndarray对象,方法是将ndarray加入到DataFrame中。其中一个ndarray是否加入DataFrame需经过条件判断,代码示例:

def demo(array1, array2, array3=None):
    if not array3:
        df = pd.DataFrame(np.column_stack((array1, array2)))
    else:
        df = pd.DataFrame(np.column_stack((array1, array2, array3)))
    return df

出问题的是条件判断语句 if not array3。if not 语句有时候是有歧义的,就像报错中指出的“具有多个元素数组的真值具有二义性”(有道翻译)。结合后面建议使用a.any()或a.all()方法,这个二义性指的是数组中“任意元素为None”和“所有元素为None”。

针对上述需求,只需将判断语句换成更明确的表述即可解决这个问题,修改后的代码如下:

def demo(array1, array2, array3=None):
    # 修改为语义更加明确的语句
    if array3 is None:
        df = pd.DataFrame(np.column_stack((array1, array2)))
    else:
        df = pd.DataFrame(np.column_stack((array1, array2, array3)))
    return df