Загрузка данных с использованием массивов в 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
Но я не могу справиться с этим. Можете ли вы посоветовать правильный подход для этого?