Porting popular R library KernSmooth to python.
Functions for Kernel Smoothing and Density Estimation.
Transformed R and Fortran functions into Python(2,3) code.
Please use kern-smooth 1.1.0 or newer. Reason: found not needed log10 density transformation.
def densCols(x, y=None, nbin=128, bandwidth=None)Produces a vector of numbers which encode the local densities at each point in dataset.
x, y : 1D numpy array with coordinates of the points density will be estimated on
nbin : [optional] int or [int, int] - number of bins along each axis (in case of single value - [nbin, nbin] will be used). Default value 128.
bandwidth : [optional] numeric vector (len of 1 or 2) of smoothing bandwidth.
Returns: numpy array with numerical representation (in range [0,1]) of point densities.
Attention: For return value numpy.nan values are allowed in case of nan / infinite values in original dataset
Source: R::grDevices::densCols
pip install kern-smooth
Make sure matplotlib is installed.
from matplotlib import pyplot as plt
from matplotlib import cm
import numpy as np
np.random.seed(0)
# create two 'bulbs' with normal distributions
mean1 = [0, 0]
cov1 = [[5, 0], [0, 30]] # diagonal covariance
x1, y1 = np.random.multivariate_normal(mean1, cov1, 50000).T
mean2 = [5, 17]
cov2 = [[30, 0], [0, 5]] # diagonal covariance
x2, y2 = np.random.multivariate_normal(mean2, cov2, 50000).T
x = np.hstack([x1,x2])
y = np.hstack([y1,y2])from kern_smooth import densCols
densities = densCols(x, y, nbin = 128)sc = plt.scatter(x, y, c=densities, s=15, edgecolors='none', alpha=0.75, cmap=cm.jet)
plt.colorbar(sc)
plt.show()
Alexander Butyaev