numpy 补充

np.where

import numpy as npimport randomx = np.arange(9).reshape(3,3)# a= x[np.where(x>5)]# b = np.where(x % 2 ==0,x ,x-1)# print(x,b)a,b = np.where(x>5)a = x[(a,b)]print(a)

np.(v/h)stack

import numpy as npa = np.array([1,2,3])b = np.array([2,3,4])np.hstack((a,b))