python输出列表中非0元素的索引，输出指定索引对应的列表中的元素

如何使用python输出列表中非0元素的索引呢？
话不多说，直接上代码：

a = [0,1,2,3]
b = [i for i, e in enumerate(a) if e != 0]
print(b)

结果：

# 列表a中非0元素的索引如下：
[1, 2, 3]

那么，如何输出指定索引对应的列表中的元素呢？

c = [a[i] for i in b]
print(c)

结果：

#列表a中非0元素的索引如下：
[3, 4, 5]

总代码如下：

a = [0,3,4,5]
b = [i for i, e in enumerate(a) if e != 0]
print(b)
c = [a[i] for i in b]
print(c)