如何从Pandas DataFrame中提取列表的列表?

Question

我在Pandas中有一个DataFrame:

import numpy as np
import pandas as pd
d = {'Person': ["A", "B", "B", "C"], 'Movies': ["ET", "Apollo 13", "12 Angry Men", "Citizen Kane"]}
df = pd.DataFrame(data=d)
print df

我想从这个DataFrame中提取一个列表的列表，其中每个人包含一个列表，并且包含他们观看的所有电影。比如:

[["ET"], ["Apollo 13", "12 Angry Men"], ["Citizen Kane"]]

这可能很简单，但我很纠结。

提前感谢!

Answer 1

尝试:

print(df.groupby("Person").agg(list)["Movies"].to_list())

打印:

[['ET'], ['Apollo 13', '12 Angry Men'], ['Citizen Kane']]