python常见处理张量库

最新推荐文章于 2025-08-15 22:59:14 发布

原创最新推荐文章于 2025-08-15 22:59:14 发布 · 174 阅读

CC 4.0 BY-SA版权

文章标签：

用于张量操作的重塑和维度变换，支持清晰的可读性操作语法。

from einops import rearrange
import numpy as np

x = np.random.rand(32, 64, 128)
y = rearrange(x, 'b c h -> b h c')  # 交换维度

PyTorch的预训练模型库，支持多种计算机视觉模型。

import timm

model = timm.create_model('resnet50', pretrained=True)

用于验证数据格式（如URL、邮箱等）。

import validators

validators.url('https://example.com')  # 返回True或False

用于终端文本颜色和样式控制。

from sty import fg, bg, ef, rs

print(fg.red + '红色文本' + fg.rs)

高效视频读取库，适用于深度学习任务。

import decord

vr = decord.VideoReader('video.mp4')
frames = vr.get_batch([0, 1, 2])  # 读取前三帧

现代HTTP客户端，支持异步请求。

import httpx

response = httpx.get('https://example.com')
print(response.status_code)

生成Excel文件的库。

import xlsxwriter

workbook = xlsxwriter.Workbook('output.xlsx')
worksheet = workbook.add_worksheet()
worksheet.write('A1', 'Hello')
workbook.close()

数据分析工具，支持表格操作。

import pandas as pd

df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
print(df.head())

数据可视化库。

import matplotlib.pyplot as plt

plt.plot([1, 2, 3], [4, 5, 6])
plt.xlabel('X轴')
plt.show()

格式化表格输出。

from tabulate import tabulate

data = [['Alice', 24], ['Bob', 19]]
print(tabulate(data, headers=['Name', 'Age']))

终端富文本渲染。

from rich.console import Console

console = Console()
console.print("[bold red]Warning![/bold red]")

文件锁工具，用于进程同步。

import portalocker

with open('file.txt', 'r+') as f:
    portalocker.lock(f, portalocker.LOCK_EX)
    f.write('Locked content')

图像读写库。

import imageio

image = imageio.imread('image.png')
imageio.imwrite('output.jpg', image)