python 拾遗

目录

“””

python import 时查找sys.path路径。

python 运行时会将本文件目录添加进sys.path中，所以可以直接import本目录的文件。

如果想import上层目录的文件，则需要将上层目录添加进sys.path

常规的做法时直接将项目的根目录添加进 sys.path 即：sys.path.insert(0,os.getcwd()), 其中 os.getcwd() 就是得到输入命令时在的目录

import sys
import os

print(sys.path)
print(os.getcwd())
sys.path.insert(0, os.getcwd())
print(sys.path)

python 项目中的所有文件中的相对路径都是以os.getcwd()为根的相对路径。

我们可以得到当前文件的绝对路径，所有的路径都已所在文件的绝对路径为基础。

pwd_path=os.path.abspath(os.path.dirname(file))

• root
  • code
    • foo.py
  • target.txt

foo.py:

pwd_path=os.path.abspath(os.path.dirname(file))
with open (“../target.txt”) as t:

否则的话就要写成：

with open(“target.txt”) as t:

“””

扁平化列表

将不等长度的列表存储至一个一维list中
“””
import itertools
a = [[1, 2, 3, 4], [4, 4, 5, 6], [7], [7, 8, 9]]
out = list(itertools.chain.from_iterable(a))

out = list(itertools.chain(*a))

print(out)
output:[1, 2, 3, 4, 4, 4, 5, 6, 7, 7, 8, 9]

“””

保留几位小数

loss=round(float(model.step(bx,by)),6)

维度交换

a=torch.rand(4,3,28,32)
a.permute(0,2,3,1).shape

torch.Size([4,28,32,3])