ustat_var.generate_test_data.generate_unique_nan_arrays

ustat_var.generate_test_data.generate_unique_nan_arrays(n_rows, n_cols, n_arrays, nan_prob, min_int, max_int, balanced=False, seed=None)

Generates unique arrays with either balanced or unbalanced NaN patterns. Each element will be a random integer.

Parameters:
  • n_rows (int) – Number of rows in each array.

  • n_cols (int) – Number of columns in each array.

  • n_arrays (int) – Number of arrays to generate.

  • nan_prob (float) – Probability of a NaN per element (when unbalanced, applies once for all arrays. See balanced argument).

  • min_int (int) – Minimum integer for random number generation.

  • max_int (int) – Maximum integer for random number generation.

  • balanced (boolean) – If True, all arrays share the same NaN pattern but have different values otherwise.

  • seed (int) – Optional random seed.

Returns:

Arrays with random numbers and NaN elements.

Return type:

list of ndarray