Python 判断数列是否为零并返回非零元素

可以使用 Python 中的列表推导式来判断并返回不为0的数列。具体实现如下：

def check_zero(lst1, lst2):
    '判断两个数列是否为0，并返回不为0的数列'
    if all(num == 0 for num in lst1) and all(num == 0 for num in lst2):
        # 如果两个数列都为0，返回空列表
        return []
    elif all(num == 0 for num in lst1):
        # 如果第一个数列为0，返回第二个数列中不为0的数
        return [num for num in lst2 if num != 0]
    elif all(num == 0 for num in lst2):
        # 如果第二个数列为0，返回第一个数列中不为0的数
        return [num for num in lst1 if num != 0]
    else:
        # 如果两个数列都不为0，返回两个数列中不为0的数
        return [num for num in lst1 + lst2 if num != 0]

使用示例：

>>> lst1 = [1, 2, 0, 3, 0]
>>> lst2 = [0, 4, 5, 0, 6]
>>> check_zero(lst1, lst2)
[1, 2, 3, 4, 5, 6]
>>> lst1 = [0, 0, 0]
>>> lst2 = [0, 0, 0]
>>> check_zero(lst1, lst2)
[]
>>> lst1 = [0, 0, 0]
>>> lst2 = [1, 2, 3]
>>> check_zero(lst1, lst2)
[1, 2, 3]
>>> lst1 = [4, 5, 6]
>>> lst2 = [0, 0, 0]
>>> check_zero(lst1, lst2)
[4, 5, 6]