Загрузка данных с использованием массивов в Python

Есть данные в таком формате в .txt файле:

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

То, что я ищу - некоторая функция, которая может дать группировку результатов для каждого идентификатора пользователя, создавая список wordid:

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

Что я пытаюсь сделать, это:

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

Но я не могу справиться с этим. Можете ли вы посоветовать правильный подход для этого?

Ответы на вопрос(1)

Ваш ответ на вопрос