Python-сообщество

Gattaka · Дек. 19, 2016 10:26:38

Коллеги,
Простой казалось бы вопрос. У меня есть массив массивов:

 [[2,3,5,6,7],
  [5,6,1,2,3]]

Я хочу из этого массива сделать мапинг, заменив числа другими. То есть 2->0, 3->1, 5->2. Чтобы в итоге получилось:

 [[0,1,2,3,4],
  [2,3,5,0,1]]

Как такое делают на питоне? Спасибо!

Отредактировано Gattaka (Дек. 19, 2016 10:27:06)

py.user.next · Дек. 19, 2016 10:39:11

  
>>> def tr(lst):
...     d = {2: 0,
...          3: 1,
...          5: 2}
...     return [d.get(i, i) for i in lst]
... 
>>> lst = [[2, 3, 5, 6, 7],
...        [5, 6, 1, 2, 3]]
>>> 
>>> list(map(tr, lst))
[[0, 1, 2, 6, 7], [2, 6, 1, 0, 1]]
>>>

Python-сообщество

Уведомления

#1 Дек. 19, 2016 10:26:38

Мапинг числа на другой набор чисел

#2 Дек. 19, 2016 10:39:11

Мапинг числа на другой набор чисел

Board footer