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