get_column_as_flat_array

pylluminator.utils.get_column_as_flat_array(df: DataFrame, column: str | list, remove_na: bool = False)

Get values from one or several columns of a pandas dataframe, and return a flatten list of the values.

Parameters:

df (pandas.DataFrame) – input dataframe
column (str | list[str]) – name of the column(s) to extract values from
remove_na (bool) – if set to True, all NaN values will be removed. Default: False

Returns:

the values as a flat list

Return type: