import numpy as np probabilities = np.array([0.2, 0.3, 0.5]) draws = np.random.default_rng(41).multinomial(n=10, pvals=probabilities, size=4) print(draws.shape) print(draws.sum(axis=1).tolist()) print(bool((draws >= 0).all()))