# String embedFile
def readEmbedFile(embedFile):
embedId = {}
input = open(embedFile,'r')
lines= []
for line in input:
lines.append(line)
nwords = len(lines)-1
splits = lines[1].split(' ') #因为第一行是统计信息,所以用第二行
dim = len(splits)-1
embeddings = [[0 for col in range(dim)] for row in range(nwords)]
for lineId in range(len(lines)):
splits = lines[lineId].split(' ')
if len(splits) > 2 :
#embedId赋值
embedId[splits[0]] = lineId
#embeddings赋值
emb = [splits[i] for i in range(1, 201)]
embeddings.append(emb)
return embedId, embeddings
关注点在于二维数组的初始化以及二维数组值的填充
更多推荐
python 读词向量文件
发布评论