📅  最后修改于: 2023-12-03 14:46:23.619000             🧑  作者: Mango
在处理Python列表时,我们可能需要找到列表中非零元素的索引位置。这在数据处理中非常常见,因为我们通常只对非零数据感兴趣。
我们可以使用Python内置的enumerate()
函数和列表推导式来实现这个需求。具体步骤如下:
enumerate()
枚举列表中的每一个元素,同时获取该元素的索引位置。以下是实现代码:
def find_nonzero_indices(lst):
return [i for i, item in enumerate(lst) if item != 0]
我们可以将代码应用在一个具体的示例中,比如找到以下列表中非零元素的索引位置:
my_list = [0, 1, 2, 0, 4, 0]
我们调用find_nonzero_indices()
函数,如下所示:
indices = find_nonzero_indices(my_list)
print(indices)
运行结果将为:
[1, 2, 4]
这表示该列表中,非零元素分别在1、2、4索引位置上。