import numpy as np
a = np.arange(10)
b = np.where(a > 5) #此时b的类型为tuple
b = np.asarray(b).reshape(-1) #asarray将b转换成二维数组,reshape(-1)将其转换为一维数组
或者(参考np.where()函数(返回值很坑)_np.where()[0]函数_小白tree的博客-CSDN博客)
import numpy as np
a = np.arange(10)
b = np.where(a > 5)[0]