def load_txt(filename):
lines=[]
vec_dict={}
with open(filename,'r') as f:
for line in f:
line=line.split(" ")
wordID=line[0]
wordvec=[float(line[j]) for j in range(1,len(line))]
vec_dict[wordID]=wordvec
return vec_dict
1.txt内容
u1 0.1 0.2 0.3
u2 0.1 0.2 0.3
u3 0.2 0.5 0.4
d=load_txt('1.txt')
结果:
{'u1': [0.1, 0.2, 0.3], 'u2': [0.1, 0.2, 0.3], 'u3': [0.2, 0.5, 0.4]}
a=list(d.values())
[[0.1, 0.2, 0.3], [0.1, 0.2, 0.3], [0.2, 0.5, 0.4]]