Carga de datos usando matrices en Python
Tener datos en dicho formato en el archivo .txt:
UserId WordID
1 20
1 30
1 40
2 25
2 16
3 56
3 44
3 12
Lo que estoy buscando: alguna función que pueda dar la agrupación de resultados para cada ID de usuario creando una lista de ID de palabra:
[[20, 30, 40], [25, 16], [56, 44, 12]]
Lo que intento hacer es:
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
Pero no puedo manejarlo. ¿Puede por favor aconsejar el enfoque correcto para hacer eso?