#NUMPY010. 查找非零元素索引

查找非零元素索引

描述

给定一个 numpy\texttt{numpy} 数组,找出其中非零元素的索引。

格式

输入

一行,一个 Python 列表形式的数组,如 [1,2,0,0,4,0]

输出

numpy\texttt{numpy} 打印的非零元素索引数组。

样例

[1,2,0,0,4,0]
[0 1 4]

提示

np.nonzero\texttt{np.nonzero}(arr) —— 返回非零元素的索引。eval(input())\texttt{eval(input())} —— 将输入字符串转换为 Python 对象。