方法1
利用dict实现构建;利用enumerate实现遍历
words=["我","北京","天安门"]
word2id={word:index for index,word in enumerate(words)}
id2word={index:word for index,word in enumerate(words)}
print(word2id)
print(id2word)
输出为:
{'我': 0, '北京': 1, '天安门': 2}
{0: '我', 1: '北京', 2: '天安门'}
方法2
利用pandas的Series实现
import pandas as pd
ids=range(len(words))
word2id=pd.Series(ids,index=words).to_dict()
id2word=pd.Series(words,index=ids).to_dict()
print(word2id)
print(id2word)
输出
{'我': 0, '北京': 1, '天安门': 2}
{0: '我', 1: '北京', 2: '天安门'}
网友评论