编程小技巧

本节仅展示一些自己碰到过的比较有意思或者比较优雅的 Python 编程技巧

字典反序:

idx_to_token = ['a','b','c']
token_to_idx = {token: idx for idx, token in enumerate(idx_to_token)}
print(token_to_idx) # 结果:{'a': 0, 'b': 1, 'c': 2}

二维数组展平为一维:

a = [[1,2],[3,4]]
b = [elem for sublist in a for elem in sublist]
print(b) # 结果:[1, 2, 3, 4]

猜你喜欢

转载自blog.csdn.net/qq_55621259/article/details/130155293