features = [] for c in income.columns: if c != income_label_col: if str(income[c].dtype) == 'category': features.append(income[c].cat.codes) else: features.append(income[c]) income_feature_vecs = numpy.array(features).T