Сортировка списка объектов по индексу свойства объекта из другого списка
У меня есть кортеж RELAY_PINS, в котором хранятся номера контактов GPIO в том порядке, в котором установлены реле. RELAY_PINS является неизменяемым, и его порядок не меняется, в то время как порядок определения устройств часто меняется.
МРЭ:
from random import shuffle, randint
class Device:
def __init__(self, pin_number):
self.pin_number = pin_number
def __str__(self):
return str(self.pin_number)
RELAY_PINS = ( 14, 15, 18, 23, 24, 25, 1, 12, 16, 20, 21, 26, 19, 13, 6, 5 )
def MRE():
devices = [ Device(pin) for pin in RELAY_PINS ]
# each GPIO pin might not have a device assigned, thus the MRE needs to emulate that
devices.pop(randint(1, len(RELAY_PINS)))
# the ordering for the list of devices should be considered random for the sake of this question
shuffle(devices)
return devices
Мое решение «работает», но, честно говоря, это смущает:
def main():
devices = MRE()
pin_map = { pin_number : index for index, pin_number in enumerate(RELAY_PINS) }
ordered_devices = [ None for _ in range(len(RELAY_PINS)) ]
for device in devices:
index = pin_map[device.pin_number]
ordered_devices[index] = device
return [ dev for dev in ordered_devices if dev is not None ]
Я знаю, что есть лучшее решение, но я не могу полностью осознать это.
Каково питоническое решение этой проблемы?
@mkrieger1, потому что я знал, что существует однострочное решение, но не видел его.
Я запустил это и сразу получил IndexError: pop index out of range.
Являются ли контакты в каждом списке уникальными или могут быть дубликаты?
Ответы 1
Вы можете использовать sorted с функцией key:
from random import randint, shuffle
class Device:
def __init__(self, pin_number: int):
self.pin_number = pin_number
def __str__(self) -> str:
return str(self.pin_number)
def __repr__(self) -> str:
return f'Device(pin_number = {self.pin_number})'
RELAY_PINS: tuple[int, ...] = (14, 15, 18, 23, 24, 25, 1, 12, 16, 20, 21, 26,
19, 13, 6, 5)
def MRE() -> None:
devices = [Device(pin) for pin in RELAY_PINS]
devices.pop(randint(0, len(RELAY_PINS) - 1))
shuffle(devices)
return devices
def main() -> None:
devices = MRE()
ordered_devices = sorted(devices, key=lambda d: RELAY_PINS.index(d.pin_number))
print(ordered_devices)
if __name__ == '__main__':
main()
Пример вывода (устройство со случайным появлением pin_number=12):
[Device(pin_number=14), Device(pin_number=15), Device(pin_number=18), Device(pin_number=23), Device(pin_number=24), Device(pin_number=25), Device(pin_number=1), Device(pin_number=16), Device(pin_number=20), Device(pin_number=21), Device(pin_number=26), Device(pin_number=19), Device(pin_number=13), Device(pin_number=6), Device(pin_number=5)]
Если сортируемый список большой, лучше всего преобразовать RELAY_PINS в словарь, который сопоставляет каждый номер контакта с его индексом, и использовать его в ключевой функции.
@Barmar Это действительно было бы лучше? Разве решение вопроса не было бы лучше?
@nocomment Я просто говорю, что это лучше, чем вызывать .index() в ключевой функции, поскольку это время O(n).
@Barmar, этот код предназначен для инструмента, который я использую 10 раз в месяц. поэтому эффективность не имеет значения
Вы здесь не единственная аудитория, этот ответ предназначен для всех, у кого схожие проблемы. @ЛордЭлронд
Что смущает в этом довольно прямолинейном и понятном коде?