import torch
import numpy as np
from scipy.stats import rankdata
a=torch.tensor([0, 2, 3, 2])
b=np.array([0, 2, 3, 2])
rankdata(a, method='dense')#tensor
rankdata(b, method='dense')#numpy都行
array([1, 2, 3, 2], dtype=int32)

该文展示了如何使用PyTorch库的torch.tensor和NumPy的np.array来实现数据排名。具体操作是调用scipy.stats的rankdata函数,采用dense方法对等值数据进行排名,结果分别是torchtensor和numpy数组。
import torch
import numpy as np
from scipy.stats import rankdata
a=torch.tensor([0, 2, 3, 2])
b=np.array([0, 2, 3, 2])
rankdata(a, method='dense')#tensor
rankdata(b, method='dense')#numpy都行
array([1, 2, 3, 2], dtype=int32)

您可能感兴趣的与本文相关的镜像
Python3.11
Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本
490
1410

被折叠的 条评论
为什么被折叠?