Tenha um dado nesse formato no arquivo .txt:

UserId   WordID
  1       20
  1       30
  1       40
  2       25
  2       16
  3       56
  3       44
  3       12

O que estou procurando - alguma função que possa fornecer o agrupamento de resultados para cada ID do usuário que cria uma lista de ID do Word:

[[20, 30, 40], [25, 16], [56, 44, 12]]

O que estou tentando fazer é:

def loadSet(path='/data/file.txt'):
  datset={}
  for line in open(path+'/file.txt'):
    (userid,wordid)=line.split('\t')
    dataset.setdefault(user,{})
    dataset[userid][wordid]=float(wordid)
    return dataset

Mas eu não posso lidar com isso. Você pode aconselhar a abordagem correta para fazer isso?