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

以下是使用Python判断两个数列是否为0，并只返回不为0的数列的代码：

def check_not_zero(list1, list2):
    if all(x == 0 for x in list1) or all(x == 0 for x in list2):
        return None
    else:
        return [x for x in list1 if x != 0], [x for x in list2 if x != 0]

这个函数接受两个数列作为参数，如果其中任意一个数列全为0，则返回None，否则返回两个数列中不为0的元素组成的元组。

使用all函数判断数列中所有元素是否都为0，如果都为0，则返回True，否则返回False。使用列表推导式将不为0的元素筛选出来，形成新的数列。

示例：

>>> check_not_zero([1, 2, 0, 3], [4, 0, 5, 0])
([1, 2, 3], [4, 5])
>>> check_not_zero([0, 0, 0], [1, 2, 3])
None
>>> check_not_zero([0, 0, 0], [0, 0, 0])
None