nispace.utils.utils.get_column_names

nispace.utils.utils.get_column_names(df_or_series, force_list=False)[source]

Get column names from a DataFrame, the name from a Series, or None if input is a numpy array.

Parameters: df_or_series (pd.DataFrame, pd.Series, or np.ndarray): The DataFrame, Series, or numpy array to get the names from.

Returns: list or None: List of column names if input is a DataFrame, str if input is a Series, or None if input is a numpy array. If force_list is True, will always return a list.