Government Transfer Distribution Estimation Functions#
transfer_distribution.py modules
ogusa.transfer_distribution#
- ogusa.transfer_distribution.MVKDE(S, J, proportion_matrix, zaxis_label='Received proportion of total transfers', filename=None, plot=False, bandwidth=0.25)[source]#
Generates a Multivariate Kernel Density Estimator and returns a matrix representing a probability distribution according to given age categories, and ability type categories.
- Parameters:
S (scalar) – the number of age groups in the model
J (scalar) – the number of ability type groups in the model.
proportion_matrix (Numpy array) – SxJ shaped array that represents the proportions of the total going to each (s,j) combination
filename (str) – the file name to save image to
plot (bool) – whether or not to save a plot of the probability distribution generated by the kde or the proportion matrix
bandwidth (scalar) – used in the smoothing of the kernel. Higher bandwidth creates a smoother kernel.
- Returns:
- SxJ shaped array that
that represents the smoothed distribution of proportions going to each (s,j)
- Return type:
estimator_scaled (Numpy array)
- ogusa.transfer_distribution.get_transfer_matrix(J=7, lambdas=array([0.25, 0.25, 0.2, 0.1, 0.1, 0.09, 0.01]), data_path=None, output_path=None)[source]#
Compute SxJ matrix representing the distribution of aggregate government transfers by age and lifetime income group.
- Parameters:
J (int) – number of lifetime income groups
lambdas (Numpy array) – length J array of lifetime income group proportions
data_path (str) – path to PSID data
output_path (str) – path to save output plots and data
- Returns:
- SxJ shaped array that represents the
smoothed distribution of proportions going to each (s,j)
- Return type:
kde_matrix (Numpy array)