记录一些有点晕的代码

np.where

看到别人的学习代码里有这个，其实没太懂，但是自己玩了一下似乎懂了

a = np.array([6,16,166,661])
b = np.array([1,5,16,14])
for i in range(len(b)):
  ind, =np.where(a>b[i])
  print(ind)
  ind = ind[0]
  print(ind)

输出满足条件的index：
[0 1 2 3]
0
[0 1 2 3]
0
[2 3]
2
[1 2 3]
1