I have the following data:How can I convert it into something like this?itertools.chain with GroupBy.transform First, get rid of NaNs inside your column using a list comprehension (messy, I know, but this is the fastest way to do it).