根據日期和第一個元素對串列進行排序

ArrayData = [['a', 'ad', '02/10/2021  7:39:19 am', 'Rank:1'],
             ['b', 'db', '02/10/2021 6:25:20 am', 'Rank:2'],
             ['a', 'sd', '02/10/2021  5:39:19 am', 'Rank:3'],
             ['b', 'esas', '02/10/2021 6:25:20 am', 'Rank:1'],
             ['a', 'aser', '02/10/2021  9:39:19 am', 'Rank:2'],
             ['d', 'ssss', '02/10/2021  11:39:19 am', 'Rank:1']]

腳本應該

1. 根據時間對同一組進行排序（例如，先對組 'a' 進行排序，然后是組 'b'、'c'、'd'）。最近的時間和日期具有更高的等級。

2. 更新每個子陣列中的“等級”

預期輸出：

[['d', 'ssss', '02/10/2021  11:39:19 am', 'Rank:1'],
 ['b', 'esas', '03/10/2021 6:25:20 am', 'Rank:2'],
 ['b', 'db', '02/10/2021 6:25:20 am', 'Rank:1'],
 ['a', 'aser', '02/10/2021  9:39:19 am', 'Rank:3'],
 ['a', 'ad', '02/10/2021  7:39:19 am', 'Rank:2'],
 ['a', 'sd', '02/10/2021  5:39:19 am', 'Rank:1']]

這是我寫的當前腳本

import operator
result = sorted(ArrayData, key=operator.itemgetter(2), reverse=True)
print(result)

我可以知道如何改進嗎？

uj5u.com熱心網友回復：

請注意，這會將您的日期時間字串轉換為datetime.datetime物件。這可能是也可能不是可取的，但如果您計劃執行涉及這些日期的任何其他操作，至少建議這樣做。如果您真的希望它們作為字串，請參閱注釋的代碼行。

另請注意，我假設您的日期是dd/mm/yyyy. 如果是mm/dd/yyyy，則需要在%d和%m中切換DATETIME_FORMAT。

import datetime
import itertools
from operator import itemgetter as get


# Assumes day/month/year, switch %d and %m if not
DATETIME_FORMAT = "%d/%m/%Y %I:%M:%S %p"


def parse_datetimes(data: list) -> list:
    result = []
    for first, second, timestamp, rank in data:
        timestamp = datetime.datetime.strptime(timestamp, DATETIME_FORMAT)
        result.append([first, second, timestamp, rank])
    return result


def custom_sort(data: list) -> list:
    # Convert datetime strings to datetime objects, then sort by first element
    sorted_data = sorted(parse_datetimes(data), key=get(0), reverse=True)

    # Re-rank each group sorted by date
    result = []
    for _, group in itertools.groupby(sorted_data, key=get(0)):
        ranked_group = []
        sorted_group = sorted(group, key=get(2))
        for rank, (*item, _) in enumerate(sorted_group, 1):
            # item[2] = item[2].strftime(DATETIME_FORMAT)
            ranked_group.append([*item, f"Rank:{rank}"])
        result.extend(ranked_group[::-1])
    return result

演示：

>>> custom_sort(ArrayData)
[['d', 'ssss', datetime.datetime(2021, 10, 2, 11, 39, 19), 'Rank:1'],
 ['b', 'esas', datetime.datetime(2021, 10, 2, 6, 25, 20), 'Rank:2'],
 ['b', 'db', datetime.datetime(2021, 10, 2, 6, 25, 20), 'Rank:1'],
 ['a', 'aser', datetime.datetime(2021, 10, 2, 9, 39, 19), 'Rank:3'],
 ['a', 'ad', datetime.datetime(2021, 10, 2, 7, 39, 19), 'Rank:2'],
 ['a', 'sd', datetime.datetime(2021, 10, 2, 5, 39, 19), 'Rank:1']]

uj5u.com熱心網友回復：

其他解決方案對我來說似乎太復雜了。如果您愿意，可以使用字串比較而無需求助于datetimes。

這是我組裝在一起的自定義解決方案，它似乎提供了所需的輸出：

from collections import defaultdict
from pprint import pprint
from typing import DefaultDict, List


array_data = [['d', 'ssss', '11-04-20', 'Rank:1'],
              ['a', 'ad', '10-13-20', 'Rank:1'],
              ['b', 'db', '12-13-20', 'Rank:2'],
              ['a', 'sd', '05-13-20', 'Rank:3'],
              ['b', 'esas', '12-14-20', 'Rank:1'],
              ['a', 'aser', '12-13-20', 'Rank:2']]


final_array = []
group_to_data: DefaultDict[str, List[List[str]]] = defaultdict(list)

for data in array_data:
    group_to_data[data[0]].append(data)


def sort_fn(x):
    """Sort by group, then by date"""
    month, day, year = x[2].split('-')
    return f'{year}{month}{day}'


for _, data in sorted(group_to_data.items(), reverse=True):
    # sorts sub-list for each group
    data.sort(key=sort_fn)

    # iterating over data in reverse order, since that's how we want it in
    # final result
    for i in range(len(data) - 1, -1, -1):
        new_rank = f"Rank:{i   1}"
        item = data[i]
        item[-1] = new_rank
        final_array.append(item)

pprint(final_array)

標籤：Python 排序

上一篇：對二維字串串列python進行排序

下一篇：奇怪的FULLOUTERJOIN"ON"條款？