如何从 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']]

如何从 Pandas DataFrame 中提取列表列表？

How do I extract a list of lists from a Pandas DataFrame?

python

list

pandas