numpy.where

>>> a=1
>>> import numpy as np
>>> b=np.array([3,5,7])
>>> i=np.where(a<b)
>>> print(i)
(array([0, 1, 2]),)
>>> a=4
>>> i=np.where(a<b)
>>> print(i)
(array([1, 2]),)
>>> i=i[0]
>>> print(i)
[1 2]
>>> i=i[0]
>>> print(i)
1
>>>

idx = np.where(epoch<lr_decay_epoch)[0][0]
--lr_decay_epoch 50 60 70


>>> lr_decay_epoch=np.array([50,60,70])
>>> epoch=49
>>> print(idx = np.where(epoch<lr_decay_epoch))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'idx' is an invalid keyword argument for this function
>>> idx = np.where(epoch<lr_decay_epoch)
>>> print(idx)
(array([0, 1, 2]),)
>>> print(idx[0][0])
0
>>> epoch=50
>>> idx = np.where(epoch<lr_decay_epoch)
>>> print(idx)
(array([1, 2]),)
>>> print(idx[0][0])
1
>>> epoch=60
>>> idx = np.where(epoch<lr_decay_epoch)
>>> print(idx)
(array([2]),)
>>> print(idx[0][0])
2
>>> epoch=70
>>> idx = np.where(epoch<lr_decay_epoch)
>>> print(idx)
(array([], dtype=int64),)
所以当大于70的时候就空了