B \#@stddlmZddlZddlmZddlZddlm Z ddl m Z ddl m Z ddgZe d dd dZdd d ZdS))divisionN)check_random_state)sample_without_replacement) deprecatedrchoicezsklearn.utils.random.choice was deprecated in version 0.19 and will be removed in 0.21. Use np.random.choice or np.random.RandomState.choice instead.TcCs6|dk r t|}|||||Stj||||SdS)a choice(a, size=None, replace=True, p=None) Generates a random sample from a given 1-D array .. versionadded:: 1.7.0 Parameters ----------- a : 1-D array-like or int If an ndarray, a random sample is generated from its elements. If an int, the random sample is generated as if a was np.arange(n) size : int or tuple of ints, optional Output shape. Default is None, in which case a single value is returned. replace : boolean, optional Whether the sample is with or without replacement. p : 1-D array-like, optional The probabilities associated with each entry in a. If not given the sample assumes a uniform distribution over all entries in a. random_state : int, RandomState instance or None, optional (default=None) If int, random_state is the seed used by the random number generator; If RandomState instance, random_state is the random number generator; If None, the random number generator is the RandomState instance used by `np.random`. Returns -------- samples : 1-D ndarray, shape (size,) The generated random samples Raises ------- ValueError If a is an int and less than zero, if a or p are not 1-dimensional, if a is an array-like of size 0, if p is not a vector of probabilities, if a and p have different lengths, or if replace=False and the sample size is greater than the population size See Also --------- randint, shuffle, permutation Examples --------- Generate a uniform random sample from np.arange(5) of size 3: >>> np.random.choice(5, 3) # doctest: +SKIP array([0, 3, 4]) >>> #This is equivalent to np.random.randint(0,5,3) Generate a non-uniform random sample from np.arange(5) of size 3: >>> np.random.choice(5, 3, p=[0.1, 0, 0.3, 0.6, 0]) # doctest: +SKIP array([3, 3, 0]) Generate a uniform random sample from np.arange(5) of size 3 without replacement: >>> np.random.choice(5, 3, replace=False) # doctest: +SKIP array([3,1,0]) >>> #This is equivalent to np.random.shuffle(np.arange(5))[:3] Generate a non-uniform random sample from np.arange(5) of size 3 without replacement: >>> np.random.choice(5, 3, replace=False, p=[0.1, 0, 0.3, 0.6, 0]) ... # doctest: +SKIP array([2, 3, 0]) Any of the above can be repeated with an arbitrary array-like instead of just integers. For instance: >>> aa_milne_arr = ['pooh', 'rabbit', 'piglet', 'Christopher'] >>> np.random.choice(aa_milne_arr, 5, p=[0.5, 0.1, 0.1, 0.3]) ... # doctest: +SKIP array(['pooh', 'pooh', 'pooh', 'Christopher', 'piglet'], dtype='|S11') N)rrnpZrandom)asizereplacep random_stater3lib/python3.7/site-packages/sklearn/utils/random.pyrs[cCs*td}td}tddg}xtt|D]}t||||<||jjdkrltd||j||jtj dd||<|dkrtj ||j dd}| d||j dnt||}t t|d std ||j d||j dkr&td |||j d|j dd||krXt||dd||<t|dd }t|} ||j ddkrd|||dk} t|| } t|| |d } || ||dk} || }|t|}t|| | }|||| ||t|q2Wtj|||f|t|ftdS)aGenerate a sparse random matrix given column class distributions Parameters ---------- n_samples : int, Number of samples to draw in each column. classes : list of size n_outputs of arrays of size (n_classes,) List of classes for each column. class_probability : list of size n_outputs of arrays of size (n_classes,) Optional (default=None). Class distribution of each column. If None the uniform distribution is assumed. random_state : int, RandomState instance or None, optional (default=None) If int, random_state is the seed used by the random number generator; If RandomState instance, random_state is the random number generator; If None, the random number generator is the RandomState instance used by `np.random`. Returns ------- random_matrix : sparse csc matrix of size (n_samples, n_outputs) irzclass dtype %s is not supportedF)copyN)shaperg?z2Probability array at index {0} does not sum to onezXclasses[{0}] (length {1}) and class_probability[{0}] (length {2}) have different length.g)Z n_population n_samplesr )dtype)arrayrangelenrZasarrayrZkind ValueErrorZastypeZint64emptyrZfillZisclosesumformatinsertrintrextendZ searchsortedZcumsumZrandappendspZ csc_matrix)rclassesZclass_probabilityr dataindicesZindptrjZ class_prob_jrngZ p_nonzeroZnnzZ ind_sampleZclasses_j_nonzeroZclass_probability_nzZclass_probability_nz_normZ classes_indrrrrandom_choice_csctsV           r&)NTNN)NN)Z __future__rZnumpyrZ scipy.sparseZsparser rZ sklearn.utilsrZ_randomrZ deprecationr__all__rr&rrrrs      _