Xarray Dataarray найти строку максимального значения

Question

Это мой первый вопрос здесь, я ценю любую помощь!

У меня есть глобальные данные в массиве данных xarray с широтой и долготой в качестве координат (разрешение .25 °), и я хотел бы найти широтудля каждого элемента долготы .25 °, где значение является самым высоким.

Я пробовал data.max('lat'), но это дает мне максимальное значение данных для каждого элемента долготы, а не широты.

Возможно, есть решение с data.where(), ноЯ еще не понял это.Так что, возможно, кто-то сделал что-то подобное и мог бы помочь мне?

Большое спасибо заранее V

Answer 1

Хорошо, нашел решение:

def findlineofmaxes(data):
    latmaxes = data.max('lat')
    lineofmaxes = data.where(data == latmaxes, drop=True)
    return lineofmaxes

Это обходной путь, но у меня работает!

Я надеюсь, что все в порядке, чтобы ответить на мой собственный вопрос, а не просто удалить его.