如何將 pandas 數據幀轉換為元組列表<string, list of></string,>

Question

我有一個具有以下結構的數據框

title      field1    field2   field3    field4   field5
title1    value11    value12  value13    value14   value15
title1    value21    value22  value23    value24   value25
title1    value31    value32  value33    value34   value35
title2    value1_1    value1_2  value1_3    value1_4   value1_5
title2    value2_1    value2_2  value2_3    value2_4   value2_5
title2    value3_1    value3_2  value3_3    value3_4   value3_5

我想將上面的數據框轉換成一個元組列表<String, List of List>，

例如，

title1, [
    (value11, value12, value13, value14, value15),
    (value21, value22, value23, value24, value25),
    (value31, value32, value33, value34, value35)
]
title2, [
    (value1_1, value1_2, value1_3, value1_4, value1_5),
    (value2_1, value2_2, value2_3, value2_4, value2_5),
    (value3_1, value3_2, value3_3, value3_4, value3_5)
]

Answer 1

您可以在 df.GroupBy.apply() 中使用 lambda function 中的tuple() df.GroupBy.apply() ，如下所示：

假設字段列從第二列開始。 如果沒有，您可以在下面的df.columns[1:]中修改列索引范圍，或者顯式使用列出下面的列名稱：

(df.groupby('title')[df.columns[1:]]
   .apply(lambda x: [tuple(y) for y in x.to_numpy()])
   .to_dict())

結果：

{'title1': [('value11', 'value12', 'value13', 'value14', 'value15'),
  ('value21', 'value22', 'value23', 'value24', 'value25'),
  ('value31', 'value32', 'value33', 'value34', 'value35')],
 'title2': [('value1_1', 'value1_2', 'value1_3', 'value1_4', 'value1_5'),
  ('value2_1', 'value2_2', 'value2_3', 'value2_4', 'value2_5'),
  ('value3_1', 'value3_2', 'value3_3', 'value3_4', 'value3_5')]}