
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)
其他用法如下:
参考:https://vimsky.com/zh-tw/examples/usage/python-scipy.stats.rankdata.html。
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)