Есть ли лучший способ заменить значения в списке Python?

На чтение 2 мин Опубликовано 17.09.2024

Вопрос или проблема

Я загружаю данные прогноза северного сияния с NOAA SWPC. Данные доступны в формате JSON, и часть из них представляет собой список списков (?), содержащих долготу, широту и значение, например:

[
   [0,-90,4],
   [0,-89,0],
   ...
]

Проблема заключается в том, что долгота в файле идет от 0 до 360, а не от -180 до 180. Чтобы это исправить, я запускаю цикл по списку:

import requests

r = requests.get("https://services.swpc.noaa.gov/json/ovation_aurora_latest.json")
data = r.json()
coords = data['coordinates']
for e in coords:
    if (e[0] > 180):
        e[0] -= 360

Есть ли “более лучший” / более “элегантный” / более “питонический” способ сделать это?

Вы можете использовать генераторы списков:

coords = data['coordinates']
coords = [[(lo - 360 if lo > 180 else lo), lat, x] for lo, lat, x in coords]

Ответ или решение

Вам абсолютно правы в том, что использование списковых включений (list comprehensions) в Python может значительно упростить код и сделать его более читаемым и понятным. Вместо цикла for вы можете использовать более элегантный способ для замены значений строго в одной строке. Вот как это может выглядеть:

import requests

# Выполняем запрос к API
r = requests.get("https://services.swpc.noaa.gov/json/ovation_aurora_latest.json")
data = r.json()

# Извлекаем координаты из полученных данных
coords = data['coordinates']

# Используем списковое включение для преобразования долгот
coords = [[(lo - 360 if lo > 180 else lo), lat, value] for lo, lat, value in coords]

В этом коде мы создаем новый список coords, где для каждого элемента lo (долгота) мы проверяем, больше ли он 180. Если да, то вычитаем 360, в противном случае оставляем его без изменений. Таким образом, каждое значение обновляется на лету, без необходимости использовать дополнительный цикл.

Такой подход не только упрощает код, но и повышает его производительность за счёт уменьшения количества строк и лишних операций. Он следует принципам функционального программирования, что делает его более "питоничным".

Если вам нужно дополнительно обработать данные или сохранить их, то это можно сделать после этого преобразования. Например, вы можете сохранить преобразованные координаты в файл или использовать их для дальнейших вычислений.

Таким образом, использование списковых включений является более изящным решением для замены значений в списках в Python.