#NUMPY034. 条件索引查找

条件索引查找

描述

给定数组和阈值 kk,找出所有大于 kk 的元素的索引。

格式

输入

两行:第一行是Python列表如 [1,2,3,4,5][1,2,3,4,5],第二行是整数 kk

输出

打印大于 kk 的元素的索引数组。

样例

[1,2,3,4,5,6,7,8,9,10]
4
[4 5 6 7 8 9]

提示

np.where\texttt{np.where}(Z > k) —— 返回满足条件的索引。