python使用技巧

本帖仅用来记录一下用法或者写法，并不讲解，如果大家都更好的用法或者写法，可以留言交流。

# 装饰器
def timespend(func):
    def wrapper(*args, **kwargs):
        start_time = time.perf_counter()
        func()
        end_time = time.perf_counter()
        print("func: {}()  use time:  {} second".format(func.__name__, (end_time - start_time)))
    return wrapper
 
# 用法
直接在函数的定义上面加 @timespend  主函数入口用不了

# tqdm 在多线程下使用姿势
import random
from time import sleep
from concurrent.futures import ThreadPoolExecutor, as_completed
from tqdm import tqdm

def func():
    sleep(random.randint(0,3))

if __name__ == '__main__':
    results = []
    with ThreadPoolExecutor(max_workers=3) as t:
        for _ in range(5):
            results.append(t.submit(func))
        for result in tqdm(as_completed(results), total=len(results), desc=func.__name__):
            pass

> func: 100%|██████████| 5/5 [00:05<00:00,  1.00s/it]

import torch
list_ = [[1,2],[2,3],[4,5]]
list_tensor = torch.tensor(list_)    #shape[3, 2]
# 如果想获取第二维度第一位的数值
two = list_tensor[:,1]     # two结果: tensor([2, 3, 5]) shape[3]
# 如果想获取第二维度第一位的数值，并将第一维度变为单维度
two = list_tensor[:,1:2]  # 或者two = list_tensor[:,1::2]
#two结果:tensor([[2],       shape[3, 1]
#               [3],
#               [5]])