Python как он есть

Question

[Роман no_name]

Допустим есть такой многомерный массив:
a= [1, 2 ,3 ,4]
[ 3, 4, 5, 6]
[5, 3, 2, 2]
. . .
есть второй массив
b= [ 1:[ab, ba]]
[2:[bc, cd]]
Нужно сделать так что бы проверяло значения первого массива и если есть совпадение дописывало его таким видом:
a= [[1,ab, ba], [2,bc, cd] ,3 ,4]
[ 3, 4, 5, 6]
[5, 3, [2, bc, cd], [2,bc, cd]]

Answer 1

[Andrey Dugin]

a = [
	[1, 2 ,3 ,4],
	[3, 4, 5, 6],
	[5, 3, 2, 2]
]

b = {
	1: ['ab', 'ba'],
	2: ['bc', 'cd']
}

for r, row in enumerate(a):
	a[r] = map(lambda x: [x]+b[x] if x in b else x, row)

print a

Answer 2

[Максим Ковалёв]

a= [[1, 2 ,3 ,4],
[ 3, 4, 5, 6],
[5, 3, 2, 2]]

b= { 1:['ab', 'ba'],
2:['bc', 'cd']}

for k,mas in enumerate(a):
....for i,j in enumerate(mas):
........c = b.get(j)
............if len(q)<=k or len(q)==0:
................q.append([])
............if c:
................q[k].append([j,c[0],c[1]])
............else:
................q[k].append(j)

Точки замени на пробелы/табы как тебе удобнее
Вот тебе быстрое решение =) Ответ получается такой:

[[[1, 'ab', 'ba'], [2, 'bc', 'cd'], 3, 4], [3, 4, 5, 6], [5, 3, [2, 'bc', 'cd'], [2, 'bc', 'cd']]]

Comments

[Роман no_name]

@kovalyov_makc а где мы q взяли?

[Andrey Dugin]

Прекрасный пример, как не надо программировать на Python.

[Максим Ковалёв]

@Romastr пардон =) q определяется перед циклом как q={}