結果 : how to search for an element in an array python