Как сделать элементы из 2d-списка капитала из данного индекса?

Question

Я хочу сделать элементы 2d-списка в капитале из индекса, возвращаемого функцией. Например:

crosswords=[['s','d','o','g'],['c','u','c','m'],['a','x','a','t'],['t','e','t','k']] 
word='cat', 

Тогда моя функция должна возвращать:

[['s','d','o','g'],['C','u','c','m'],['A','x','a','t'],['T','e','t','k']] 

Вот что я пробовал:

def find_word_vertical(crosswords, word): 
    x=zip(*crosswords) 
    #print x 
    for row in x: 
     join_list = "".join(row) 
     if join_list.find(word) != -1: 
      return [row.index(word[0]),x.index(row)] 
      break 
    def find_word_horizontal(crosswords, word): 

    for row in crosswords: 
     join_list = "".join(row) 
     if join_list.find(word) != -1: 
      return [crosswords.index(row), row.index(word[0])] 
      break 

def capitalize_word_in_crossword(crosswords,word): 

    if find_word_horizontal or find_word_vertical is not None: 
     for i in word: 
      for j in crosswords: 
       if i in j: 
        i.upper() 
        return crosswords 

Вот что результат:

[['s', 'd', 'o', 'g'], ['c', 'u', 'c', 'm'], ['a', 'c', 'a', 't'], ['t', 'e', 't', 'k']] 

Может кто-нибудь мне помогает?

Answer 1

Вы можете найти горизонтальные и вертикальные вхождения, используя понимание списка.

Горизонтальные вхождения являются:

>>> filter(
    lambda (_, j): j > -1, 
    [(i, ''.join(w).find(word)) for i, w in enumerate(crosswords)]) 
[] 

Вертикальные вхождений:

>>> filter(
    lambda (i, _): i > -1, 
    [(''.join(w[j] for w in crosswords).find(word), j) for j in range(len(crosswords))]) 
[(1, 0), (1, 2)] 

С вхождений, легко выбрать, что делать с ними (выбрать на или все, предпочитают горизонтально по отношению к вертикаль и т. д.), включая их капитализацию.