login_ids = [1, 2, 3, 4, 5] # ログインユーザー一覧 paid_ids = [2, 3, 4] # プレミアムユーザー一覧 block_ids = [1, 2] # ブロックされたユーザー一覧 login_ids = set(login_ids) paid_ids = set(paid_ids) block_ids = set(block_ids) def solve(login_ids,paid_ids,block_ids): return (login_ids and paid_ids) - block_ids ans = [] for id in login_ids: if id in block_ids: continue if id in paid_ids: ans.append(id) return ans # [[1, 2, 3], [4, 5, 6], [7, 8, 9]] # => [[1, 3], [4, 6], [7, 9]] x = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] ans = [] for ele in x: ans.append([ele[0],ele[2]]) [[True, False, Ture], [True, False, Ture], [True, False, Ture]] x.np.where(index != 1) x[:,1:] x[:,[0,2]]