pydit.wrangling.various.create_test_dataframe

pydit.wrangling.various.create_test_dataframe(dataset_name: str, n_rows: int = 10, n_cols: int = 10)[source]

Create test dataframes

IMPORTANT these are NOT to be used for the core test suite. These are meant for the examples and for users to play, tutorials etc.

Parameters:
  • dataset_name (str) – Name of the dataset One of: random_numeric, random_categorical, dataset_01

  • n_rows (int, optional, default 10) – Number of rows in the dataframe

  • n_cols (int, optional, default 10) – Number of columns in the dataframe

Returns:

A test dataframe

Return type:

pd.DataFrame