使用 Python 創建列表列表

Question

具有如下數據框：需要轉換為列表列表如下：

產品1	產品2	時間
蘋果	蘋果	0
蘋果	芒果	20
蘋果	橙子	24
芒果	蘋果	30
芒果	芒果	0
芒果	橙子	24
橙	蘋果	12
橙	橙	0
橙	芒果	24

需要創建一個邏輯如下的矩陣：

output 需要是格式列表如下：

[[0,24,20],[12,0,24],[30,24,0]]

我嘗試了以下代碼：

df.groupby(['product1','product2'])['time'].apply(list)

Answer 1

從這個看起來，這被稱為 pivot 表。 您可以使用pandas.pivot_table方法創建它。

要創建上述矩陣，您需要編寫此代碼。

pt = data.pivot_table(index="product1", columns="product2", values="time")

以列表形式獲取此列表。 將其存儲在變量中並使用它。

pt.values.tolist()