Question

我有一个包含整数的列表，想要替换它们，以便之前包含最高数字的元素现在包含1，第二个最高数字设置为2，等等。

实施例： [4, 3, 1, 6, 2, 5]应该产生$("#dialog").dialog({ autoOpen: false, height: 350, width: 530, modal: true, open: function (event, ui) { $(this).load("ChildView"); }, close: function (event, ui) { $("#dialog").dialog().dialog('close'); } }).on("ChildButtonClick", function (arg) { alert( "Hello " + arg); })。

我个人只关心前9个最高的数字我认为可能有一个简单的算法，甚至可能是python函数来处理这个任务？

编辑：我不在乎如何处理重复项。

Answer 1

快速执行此操作的方法是首先生成元素的元组列表及其位置：

sort_data = [(x,i) for i,x in enumerate(data)]

接下来我们会在reverse中对这些元素进行排序：

sort_data = sorted(sort_data,reverse=True)

生成（用于您的样本输入）：

>>> sort_data
[(34, 2), (9, 4), (6, 1), (5, 0), (3, 5), (1, 3)]

和nest我们需要填写这些元素，如：

result = [0]*len(data)
for i,(_,idx) in enumerate(sort_data,1):
    result[idx] = i

或者把它放在一起：

def obtain_rank(data):
    sort_data = [(x,i) for i,x in enumerate(data)]
    sort_data = sorted(sort_data,reverse=True)
    result = [0]*len(data)
    for i,(_,idx) in enumerate(sort_data,1):
        result[idx] = i
    return result

此方法适用于 O（n log n），其中 n data中的元素数。

更紧凑的算法（在没有为排序构造元组的意义上）是：

def obtain_rank(data):
    sort_data = sorted(range(len(data)),key=lambda i:data[i],reverse=True)
    result = [0]*len(data)
    for i,idx in enumerate(sort_data,1):
        result[idx] = i
    return result

Answer 2

如果您没有任何重复项，以下列表理解将会：

lst = [5, 6, 34, 1, 9, 3]
tmp_sorted = sorted(lst, reverse=True)  # kudos to @Wondercricket
res = [tmp_sorted.index(x) + 1 for x in lst]  # [4, 3, 1, 6, 2, 5]

要了解它是如何工作的，你可以把它分成几块：

lst = [5, 6, 34, 1, 9, 3]
# let's see what the sorted returns
print(sorted(lst, reverse=True))  # [34, 9, 6, 5, 3, 1]
# biggest to smallest. that is handy.
# Since it returns a list, i can index it. Let's try with 6
print(sorted(lst, reverse=True).index(6))  # 2
# oh, python is 0-index, let's add 1
print(sorted(lst, reverse=True).index(6) + 1)  # 3
# that's more like it. now the same for all elements of original list

for x in lst:
    print(sorted(lst, reverse=True).index(x) + 1)  # 4, 3, 1, 6, 2, 5

# too verbose and not a list yet..
res = [sorted(lst, reverse=True).index(x) + 1 for x in lst]
# but now we are sorting in every iteration... let's store the sorted one instead
tmp_sorted = sorted(lst, reverse=True)
res = [tmp_sorted.index(x) + 1 for x in lst]

Answer 3

另一种选择，您可以使用scipy中的rankdata功能，并提供处理重复项的选项：

from scipy.stats import rankdata

lst = [5, 6, 34, 1, 9, 3]
rankdata(list(map(lambda x: -x, lst)), method='ordinal')
# array([4, 3, 1, 6, 2, 5])

Answer 4

使用numpy.argsort：

numpy.argsort返回对数组进行排序的索引。

>>> xs = [5, 6, 34, 1, 9, 3]

>>> import numpy as np
>>> np.argsort(np.argsort(-np.array(xs))) + 1
array([4, 3, 1, 6, 2, 5])

Answer 5

使用纯Python的简短的对数线性解决方案，没有查找表。

想法：将位置存储在成对列表中，然后对列表进行排序以重新排序位置。

enum1 = lambda seq: enumerate(seq, start=1)  # We want 1-based positions

def replaceWithRank(xs):
    # pos = position in the original list, rank = position in the top-down sorted list.
    vp = sorted([(value, pos) for (pos, value) in enum1(xs)], reverse=True)
    pr = sorted([(pos, rank) for (rank, (_, pos)) in enum1(vp)])
    return [rank for (_, rank) in pr]

assert replaceWithRank([5, 6, 34, 1, 9, 3]) == [4, 3, 1, 6, 2, 5]

如何用（python）列表中的顺序替换数字

5 个答案: