使用 Python 列表推导式去除两个数列中的零元素

假设有两个数列,其中一个为零数列,需要将两个数列合并,并去除其中的零元素,只保留非零元素。可以使用列表推导式来实现这个功能,具体代码如下:

num_list1 = [0, 1, 0, 2, 0, 3, 4, 5, 0]
num_list2 = [6, 0, 7, 0, 8, 9, 0, 10, 0]

non_zero_list = [num for num in num_list1 + num_list2 if num != 0]
print(non_zero_list)

输出结果为:

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

解释:

首先,将两个数列 num_list1num_list2 使用 + 操作符合并成一个新的数列。

然后,使用列表推导式 [num for num in num_list1 + num_list2 if num != 0] 从合并后的数列中筛选出不为零的元素。

最后,将筛选后的非零元素组成新的数列 non_zero_list 并输出。

总结:

列表推导式是一种简洁高效的方式,可以方便地对列表进行操作,例如筛选、转换和创建新的列表。在处理类似去除零元素这类任务时,列表推导式可以提供简洁且易读的代码。


原文地址: https://www.cveoy.top/t/topic/oogB 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录