从数组[1, 2, 0, 0, 4, 0]中找出非0元素的位置索引。
时间: 2023-09-25 22:10:12 浏览: 528
好的,以下是Python的代码实现:
```python
arr = [1, 2, 0, 0, 4, 0]
# 使用列表推导式生成非0元素的位置索引列表
index_list = [i for i in range(len(arr)) if arr[i] != 0]
print(index_list)
```
这段代码中,我们使用列表推导式生成非0元素的位置索引列表。具体来说,我们使用`range`函数生成索引的范围(即从0到数组长度减1),然后使用`if`语句判断元素是否为0。如果元素不为0,则将该元素的索引添加到结果列表中。
最后,使用`print`函数输出结果。在这个例子中,输出的结果应该是`[0, 1, 4]`,对应的是数组中非0元素1、2、4的位置索引。
阅读全文