filter array based on another array python