Как лучше применить функцию к элементам вложенного списка в Python?

Question

[Александр Рублев]

Всем привет.

Не могу придумать, как быстро и красиво применять функцию к определенному элементу вложенного списка.

Например, имеем список вида имя, пол, год рождения:

[
    ['Саша', 'М', 1993],
    ['Маша', 'Ж', 1990],
    ['Паша', 'М', 1995],
    # и т.д.
]

Допустим, нам необходимо посчитать сколько лет каждому человеку и на выходе получить следующий список:

[
    ['Саша', 'М', 26],
    ['Маша', 'Ж', 29],
    ['Паша', 'М', 24],
    # и т.д.
]

То есть, нам нужно к каждому 3 элементу вложенного списка применить функцию рассчёта возраста. И перезаписать это значение.

Я знаю как это сделать через цикл for, но получается как-то не аккуратно, и не по python way. Так же меня беспокоит вопрос производительности на больших списках.

Я знаю как решить эту задачу простым решением в лоб. Но мне интересно, есть ли какой нибудь python way способ.

Буду рад любым советам. Спасибо.

Comments

[Animkim]

Когда будет не быстро, тогда и будешь думать как сделать быстро.

[Александр Рублев]

Что за бред. Ты мне предлагаешь сейчас сделать как попало, что бы потом с ростом кол-ва данных, когда это будет уже в работе, брать и переписывать? А так же бегать по всему коду и вспоминать, где я еще делал не правильно?
Вместо того, что бы заранее спросить у знающих людей совета?

[AWEme]

Александр Рублев, покажи что уже сам написал.

[Animkim]

Александр Рублев, какое количество данных у тебя сейчас? При каком количестве начнет тормозить? За сколько лет ты приблизишься к пороговому значению?
Плюс я нигде не писал что надо написать плохо, но обычного for я подозреваю тебе хватит за глаза. Или чтобы красивее то list comprehension.

[Александр Рублев]

AWEme, Например так

input = [
    ['Саша', 'М', 1993],
    ['Маша', 'Ж', 1990],
    ['Паша', 'М', 1995],
]

def calc_year(year):
    return 2019 - year

for row in input:
    row[2] = calc_year(row[2])

Я знаю как решить эту задачу простым решением в лоб. Мне интересно есть ли какой нибудь питоновский способ.

[longclaps]

Александр Рублев, молодец, чо. Ты только не обижайся )))
Скажи, а вот этот способ упихивать данные, вот этот разнородный список ['Саша', 'М', 1993] - он точно сделан не как попало?

[Александр Рублев]

longclaps,
Ну во первых это пример,
Во вторых примерно такие данные, я получаю из базы данных, Которые после обработки нужно вставить в таблицу и показать пользователю.
И честно говоря не понимаю почему они прям разнородные, пускай это будет списком работников, или списком учащихся, только в таком случае данных будет еще больше.

[longclaps]

Александр Рублев, ну во вторых это не так - из базы обычно возвращают кортежи, а не списки.
А во-первых, этот пример стоит обрабатывать именно в цикле. И, кстати, на каждый чих заводить и вызывать функцию - это двигаться ровно в обратную сторону от ускорения )))

[Александр Рублев]

longclaps, Во вторых, я согласен. Обычно кортежи.
Во первых: Понял буду обрабатывать в цикле. Опять таки это пример, на деле все немного сложнее. и списки больше, и задачи немного по интереснее, но учту это. Спасибо!
Вопрос задавался с целью просвещения, как когда то для меня просто поразило решение этого человека. До сих пор пользуюсь, сам бы не догадался.

Если не сложно, подскажите, а что делать если и вправду, входной список будет кортежем? Обрабатывать в цикле, создавая новый список?

[longclaps]

если и вправду, входной список будет кортежем?

Александр Рублев, В питоне такого не бывает ) В руби, например, можно заморозить массив - и он станет из питонячьего листа кортежем (примерно), но в питоне лист не может быть кортежем - можно лишь создать из лист с данными из кортежа и наоборот.

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

Поразившее тебя решение содержит ошибку, жаль, что ты её не заметил )

for i in a + b:
    c.setdefault(i[0], []).extend(i[1:])

[Александр Рублев]

longclaps, Спасибо за ответ. Да ошибку я не увидел, но у меня и так все хорошо работает))

Answer 1

[Сергей Горностаев]

Ваша задача решается только перебором, поэтому быстрее, чем циклом сделать не получится. К тому же, for - это вполне питонично.

Другой вариант - это использовать списковое включение для генерации нового списка на основе исходного:

a = [
  ['Саша', 'М', 1993],
  ['Маша', 'Ж', 1990],
  ['Паша', 'М', 1995],
 ...
]
current_year = date.today().year
b = [[n, s, current_year - b] for n, s, b in a]

Или map для той же цели, но вот он как раз совсем не "по питоновски"

b = list(map(lambda i: [i[0], i[1], current_year - i[2]], a))

Или чуточку покрасивее с starmap:

b = list(starmap(lambda n, s, b: [n, s, current_year - b], a))

Comments

[Александр Рублев]

Понял. Спасибо. Я как раз думал об этих способах (или нечто похожее). Надеялся что я чего то не знаю.
Оставим перебор через for