Dataframe数值转为二维列表

# Dataframe数值转为二维列表
# is_del_duplicates用来判断是否删除dataframe中重复的行（重复行中只保留第一次出现的）
# 具体可以根据自己的需要修改 drop_duplicates()中的参数
def dataframe_to_list(df, is_del_duplicates):  
    if not is_del_duplicates:  # 不删除的情况
        # 返回值是一个二维列表
        return df.values.tolist()
    else:  # 删除重复行的情况
        return df.drop_duplicates().values.tolist()