그룹화 같은이 경우 목록의 요소를 효율적으로

Question

n,m=map(int,input().split()) 
arr=[i%m for i in (map(int,(input().split())))] 

가정하자 n=5 및 m =3 등 입력 array =[3, 2 ,1 ,4, 5]arr=[0, 2, 1, 1, 2]하지만 지금은 효율적으로 [1,1] 및 [2,2] 뭐야이 그룹에 가장 좋은 방법, 즉 목록에 동등한 가치의 요소를 저장할 ? 그들을 함께 효율적으로 또한 나는 끝의 인덱스를 원하는 정도로

출력 [[1,1], [2,2] 인덱스 (2,3) 및 인덱스에서 (1,4)

내가 뭘 찾고 원래 배열 요소의 색인 mod tha를 복용하기 전에입니다 mod 연산을 수행 한 후 동일한 값을가집니다.

Answer 1

set에만 고유 한 값

arr=[0, 2, 1, 1, 2] 
arr = [(s, arr.count(s)) for s in set(arr)] 
# [(0, 1), (1, 2), (2, 2)] 

업데이트 (@JonClements 덕분에)

s = {} 
for i, v in enumerate(arr): 
    s.setdefault(v % 3, []).append(i) 
print(s) 
# {0: [0], 1: [2, 3], 2: [1, 4]}