Julia: Comment rendre colwise retourner un tableau au lieu d'un tableau de tableaux?

Question

Je voulais faire une fonction qui regarde chaque colonne d'un DataFrame et retourne un booléen, donc je me retrouve avec un tableau de booléens. Voici le code

# some random dataframe 
df = DataFrame([1:3, 4:6]) 

# a function that returns an array of boolean 
function some_bool_fn(df)::Array{Bool} 
    array_of_arrays = colwise(df) do sdd3 
     # for illustration only 
     return true 
    end 
    array = [a[1] for a in array_of_arrays] 
    return array 
end 

# calling the function 
some_bool_fn(dd3) 

Cela fonctionne sauf que je trouve la ligne

 array = [a[1] for a in array_of_arrays] 

un peu inutile. Fondamentalement, je reçois un tableau de tableaux comme la sortie de colwise, donc j'ai dû mettre le tableau de tableaux dans un tableau simple de bools. Y at-il un moyen d'écrire le code afin que je puisse éviter cette ligne de code?

Answer 1

Comme @Gnimuc a commenté ce comportement est en train de changer.

Si vous regardez la branche principale: https://github.com/JuliaData/DataFrames.jl/blob/master/src/groupeddataframe/grouping.jl#L241 vous verrez une autre version. Vous pouvez probablement le copier:

mycolwise(f, d::AbstractDataFrame) = [f(d[i]) for i in 1:ncol(d)]