본문 바로가기

Computer/Python

nested list comprehesion in python

파이썬에서 리스트를 사용할 때, list comprehesion을 통해서 표현하곤 한다.


예를 들자면...


for x in range(10)

     count(x)


이것을 list comprehension으로 표현하면...


[count(x) for x in range(10)]


이런 식으로 표현을 하는 것인데...


중첩 리스트를 어떻게 표현해야 하나 헷갈리는 경우가 많다.(나는... 그렇다.)


중첩 리스트(nested list)는 다음과 같이 표현하면 된다.


for i in range(10):

    for j in range(i):

        count(i, j)


=>


[[count(i, j) for j in range(i)] for i in range(i)]


혹시 if문을 쓰고 싶다면...


[[count(i, j) for j in range(i) if i > 5] for i in range(i)]



이런식으로 사용하면 된다.

오케이3


참고 : http://stackoverflow.com/questions/18072759/python-nested-list-comprehension

'Computer > Python' 카테고리의 다른 글

inline if  (0) 2015.03.11
파이썬 json.dumps를 한글에 사용하는 방법  (0) 2015.02.01
nested list comprehesion in python  (0) 2015.01.10
python OrderedDict  (0) 2014.12.04
파이썬 표준 에러(stderr) 출력  (0) 2014.11.21
python2.x에서 unicodedata 사용법  (0) 2014.10.21