#NUMPY010. 查找非零元素索引
查找非零元素索引
描述
给定一个 数组,找出其中非零元素的索引。
格式
输入
一行,一个 Python 列表形式的数组,如 [1,2,0,0,4,0]
输出
打印的非零元素索引数组。
样例
[1,2,0,0,4,0]
[0 1 4]
提示
(arr) —— 返回非零元素的索引。 —— 将输入字符串转换为 Python 对象。
给定一个 numpy 数组,找出其中非零元素的索引。
一行,一个 Python 列表形式的数组,如 [1,2,0,0,4,0]
numpy 打印的非零元素索引数组。
[1,2,0,0,4,0]
[0 1 4]
np.nonzero(arr) —— 返回非零元素的索引。eval(input()) —— 将输入字符串转换为 Python 对象。