Skip to content
Minsing Log
Go back

English Edition

Flattening

Programming & CS

Situation

I solved the task of flattening a pandas dataframe containing list elements into a list using list comprehension. However, this abuses list comprehension and results in code that is much less readable.

[passage for lst_passage in make_passages['passages'] for passage in lst_passage]

Solutions

  1. Use itertools
import itertools

passages = list(itertools.chain.from_iterable(make_passages['passages']))
  1. Unpacking (using *)
passages = list(*passage for passage in make_passages['passages'])
  1. Functions
def flatten_list(nested_list):
   return [item for sublist in nested_list for item in sublist]

passages = flatten_list(make_passages['passages'])

공개 여정 구독

글이 올라오면 이메일로 보내드릴게요.

개발, 창업, 배움, 실패와 회고를 짧고 솔직하게 기록합니다. 검색으로 스쳐가는 독자가 아니라 여정을 따라오는 팬을 만들기 위한 뉴스레터입니다.

뉴스레터는 준비 중입니다. 그동안 RSS 로 구독하거나 이메일 로 알림을 요청해 주세요.


Share this post:

Comments

Sign in with GitHub to leave a comment.


Previous Post
Creating multiple dataframes with apply
Next Post
apply function