여러 리스트 중 원하는 특정 원소가 있는 리스트만 골라내고 싶습니다

조회수 441회
a_list = [ '짜장면' , '짬뽕' ]
b_list = [ '볶음밥' , '마파두부' ]
c_list = [ '초밥', '튀김' ]
...
z_list = [ '어떤' , '원소']

def example():
    ~~~

위 처럼 여러개의 리스트가 있을때

example('짬뽕')
[ '짜장면', '짬뽕' ] 출력
example('마파두부')
[ '볶음밥' , '마파두부' ] 출력
example('튀김')
[ '초밥' , '튀김' ] 출력

이렇게 작동되는 코드를 짜고 싶습니다.

2 답변

  • all_list = [['a', 'b', 'c'], ['a', 'd', 'e'], ['b','g','e']]
    
    def example(word):
        l = []
        for lst in all_list:
            if word in lst:
                l.append(lst)
        return l
    print(example('a'))  # [['a', 'b', 'c'], ['a', 'd', 'e']]
    print(example('e'))  # [['a', 'd', 'e'], ['b', 'g', 'e']]
    
    • (•́ ✖ •̀)
      알 수 없는 사용자
  • a_list = [ '짜장면' , '짬뽕' ]
    b_list = [ '볶음밥' , '마파두부' ]
    all_list = []
    all_list.append(a_list)
    all_list.append(b_list)
    
    def example(_txt):
        for i in range(0,len(all_list)):
            for j in range(0, len(all_list[i])):
                if _txt == all_list[i][j]:
                    return all_list[i][j]
    
    print(example('짬뽕'))
    '''
    ===============
    짬뽕
    '''
    

    원코드를 해치지 않는선에서 해봤어요 저도 코알못이라

답변을 하려면 로그인이 필요합니다.

프로그래머스 커뮤니티는 개발자들을 위한 Q&A 서비스입니다. 로그인해야 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)