new_data = data.groupby("sku", as_index=False)["result_skus"].apply(list) new_data.head()