我在 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"]]
這可能很簡單,但我正在努力解決它。
提前致謝!
uj5u.com熱心網友回復:
嘗試:
print(df.groupby("Person").agg(list)["Movies"].to_list())
印刷:
[['ET'], ['Apollo 13', '12 Angry Men'], ['Citizen Kane']]
轉載請註明出處,本文鏈接:https://www.uj5u.com/qianduan/466040.html
上一篇:如何將一個串列中的一個元素與另一個串列的每個元素一起傳遞給一個接受兩個引數的函式?
下一篇:如何對嵌套串列進行兩次切片?
