NumPy: применение функции, которая принимает список в качестве аргумента к списку списков и получает список

Question

У меня проблема, и чтение do c не помогает. Итак, по сути, у меня есть функция, которая принимает в качестве аргумента список NumPy и возвращает число. (Например, как np.average). Назовем эту функцию f.

Теперь у меня есть список списков L. Итак, L[0] - это список, L[1] - это список ... И я хочу использовать NumPy parallelism, чтобы применить f к L и получить в результате [f(L[0]), f(L[1]),...]. Вот и моя проблема: когда я звоню f(L), результат - номер f(flattened L).

Я понятия не имею, как это исправить, может ли кто-нибудь мне помочь?

Спасибо, Kal8578

Answer 1

А как насчет преобразования f в рекурсивную функцию? Что-то вроде:

def f(x):
    if isinstance(x, list):
        return [f(y) for y in x]
    # return what you previously wrote here