python - 要添加到新列表 Python/Django 中的特定元素排序

Question

我有这个:

[
[{ 'position': 1,  'user_id': 2,  'value': 4, 'points': 100}],
[{ 'position': 2,  'user_id': 6,  'value': 3, 'points': 88}],
[{ 'position': 3,  'user_id': 5,  'value': 2, 'points': 77}],
[{ 'position': 4,  'user_id': 7,  'value': 1, 'points': 66}],
[{ 'position': 5,  'user_id': 3,  'value': 1, 'points': 9}],
[{ 'position': 6,  'user_id': 11, 'value': 0,  'points': 9}],
[{ 'position': 7,  'user_id': 1,  'value': 0, 'points': 3}],
[{ 'position': 8,  'user_id': 10, 'value': 0,  'points': 3}],
[{ 'position': 9,  'user_id': 4,  'value': 0, 'points': 2}],
[{ 'position': 10,  'user_id': 8, 'value': 0,  'points': 2}]
]

按点组织。 这个想法是选择 user_id 并生成一个包含所选 5 个用户的新列表。

示例:

user_id=3:

[{ 'position': 3,  'user_id': 5,  'value': 2, 'points': 77}],
[{ 'position': 4,  'user_id': 7,  'value': 1, 'points': 66}],
[{ 'position': 5,  'user_id': 3,  'value': 1, 'points': 9}],
[{ 'position': 6,  'user_id': 11, 'value': 0,  'points': 9}],
[{ 'position': 7,  'user_id': 1,  'value': 0, 'points': 3}]

它返回中间的 user_id 3，其中 2 个用户高，2 个用户低

user_id=2

[{ 'position': 1,  'user_id': 2,  'value': 4, 'points': 100}],
[{ 'position': 2,  'user_id': 6,  'value': 3, 'points': 88}],
[{ 'position': 3,  'user_id': 5,  'value': 2, 'points': 77}],
[{ 'position': 4,  'user_id': 7,  'value': 1, 'points': 66}],
[{ 'position': 5,  'user_id': 3,  'value': 1, 'points': 9}],

由于 user_id 没有较高的用户，因此返回 4 个较低的用户。所以总是相同的逻辑。

user_id=9:

[{ 'position': 6,  'user_id': 11, 'value': 0,  'points': 9}],
[{ 'position': 7,  'user_id': 1,  'value': 0, 'points': 3}],
[{ 'position': 8,  'user_id': 10, 'value': 0,  'points': 3}],
[{ 'position': 9,  'user_id': 4,  'value': 0, 'points': 2}],
[{ 'position': 10,  'user_id': 8, 'value': 0,  'points': 2}]

在 user_id=9 上，我们只有 1 位较低的用户，因此我们添加了 3 位较高的用户

例如，如果列表中只有 2 个用户，则它应该返回这 2 个用户。 主要规则: 如果我们有 5 个或更多用户，则返回 5 个用户。 如果我们有 4 个用户，则返回 4 个用户

有什么好的方法吗？ 谢谢

Answer 1

这基本上只是 my answer 的更新至 your original question .

a = [
    [{ 'position': 1,  'user_id': 2,  'value': 4, 'points': 100}],
    [{ 'position': 2,  'user_id': 6,  'value': 3, 'points': 88}],
    [{ 'position': 3,  'user_id': 5,  'value': 2, 'points': 77}],
    [{ 'position': 4,  'user_id': 7,  'value': 1, 'points': 66}],
    [{ 'position': 5,  'user_id': 3,  'value': 1, 'points': 9}],
    [{ 'position': 6,  'user_id': 11, 'value': 0,  'points': 9}],
    [{ 'position': 7,  'user_id': 1,  'value': 0, 'points': 3}],
    [{ 'position': 8,  'user_id': 10, 'value': 0,  'points': 3}],
    [{ 'position': 9,  'user_id': 4,  'value': 0, 'points': 2}],
    [{ 'position': 10,  'user_id': 8, 'value': 0,  'points': 2}]
]

# Sort it if not already sorted
# a.sort(key=lambda x: x[0]['position'])

def find_index(l, user_id):
    i = 0
    while l[i][0]['user_id'] != user_id:
        i += 1
    return i

def get_subset(l, i):
    return l[:(i + 1 + max(2, 4 - i))][-5:]

get_subset(a, find_index(a, 3))