[代码笔记]python函数方法笔记

发表于 收录于
一些python中的笔记

tensorboard

1
2
3
4
5
6
7
8
9
10
11
12
13
from torch.utils.tensorboard import SummaryWrite

write = SummaryWrite(log_dir="logs") # logs为自定义文件夹名字

# 一个表格中只有一条数据
write.add_scalars("名称", y轴数值， x值数值)

# 多条数据对比
write.add_scalars("总名称", {"数据名称": 数值,
                            "数据名称": 数值"}, x轴数据)

# 关闭
write.close()

启动tensorboard：控制台中输入
tensorboard --logdir=logs # logs为路径

多维数组中根据一维数组寻找其下标（numpy）

给一个多维数组nums，返回里面的一维数组target的下标index。

1
2
3
4
5
6
7
8
9
10
11
import numpy as np

nums = np.array([[1, 2, 3],
              [1, 3, 5],
              [1, 5, 6]])

target = [1, 3, 5]  

for  index, num in enumerate(nums):
    if (target == num).all():
        print(index, num)

out:
1
1 [1 3 5]