Feature Selection Code In Python For Prime