Python Feature Extraction