python - 如何从 pandas DataFrame 生成结果列表

Question

我需要从 pandas DataFrame 生成一个列表。我能够打印结果，但我不知道如何将其转换为列表格式。我用来打印结果(不转换为列表)的代码是

df=pandas.DataFrame(processed_data_format, columns=["file_name", "innings", "over","ball", "individual ball", "runs","batsman", "wicket_status","bowler_name","fielder_name"])**  
  #processed_data_format is the list passing to the DataFrame   
t = df.groupby(['batsman','over'])['runs','ball'].sum()
print t

我得到的结果是

Sangakara 1   10   5

          2    0   2

          3    3   1


    sewag 1    2   1

          2    1   1 

我想将这些数据转换成列表格式

[ [sangakara,1,10,5],[sangakara,2,0,2],[sangakara,3,3,1],[sewag,1,2,1][ sewag,2,1,1] ]

Answer 1

您可以使用 to_records :

import pandas as pd

df = pd.DataFrame({'a': [1, 2, 3, 4], 'b': [2, 3, 4, 5]})

>>> list(df.to_records())

 [(0, 1, 2), (1, 2, 3), (2, 3, 4), (3, 4, 5)]

这将生成一个元组列表。将其转换为列表列表(如果您真的需要它)很容易。