用np的where函数啊

import numpy as np
a = np.array([1,2,3])
b = np.array([True, False, True])
c = np.where(b, a, 0)
--
FROM 131.107.8.*