small trick in python

1. 截取列

tmp = torch.tensor([[1,2,3,4], [5,6,7,8]])
cut = tmp[:, :2]

截取tensor前面两列

2. 对list每项做相同操作

tnsbs = f.readlines()[1:]

tnsbs每项后面都有一个换行符’\n’, 对每个元素都去除最后一项，即换行符

tnsbs = list(map(lambda x:x[:-1], f.readlines()[1:]))

3. defaultdict

from collections import defaultdict
# 如果d[0]=0, 默认是没有这个key值的

4. copy.deepcopy()

python 中 =（等号）是使两个变量共享同一个内存地址，所以这两个变量并不相互独立。如果要相互独立，需要使用copy模块（ copy.copy() 或 copy.deepcopy() ）