numpy 补充

np.where

import numpy as np
import random
x = 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 np
a = np.array([1,2,3])
b = np.array([2,3,4])
np.hstack((a,b))