np.argwhere()相关

np.argwhere(x <= 10).flatten()

• np.argwhere(a)
  返回非零数组的索引，其中x是索引数组的条件
  举例：

import numpy as np
x=np.arange(8)
x

x1=x.reshape(2,4)
x1

np.argwhere(x>2)

• flatten是numpy.ndarray.flatten的一个函数
  即返回一个折叠成一维的数组。但是该函数只能适用于numpy对象，即array或者mat，普通的list列表是不行的。

x1.flatten()