Skip to content

Instantly share code, notes, and snippets.

@Keiku
Created February 8, 2017 02:56
Show Gist options
  • Save Keiku/b1e3837eb3ce104508637eeaf2e22862 to your computer and use it in GitHub Desktop.
Save Keiku/b1e3837eb3ce104508637eeaf2e22862 to your computer and use it in GitHub Desktop.
Cut a variable with pandas.
import pandas as pd
from sklearn import datasets
iris = datasets.load_iris()
iris_df = pd.DataFrame(iris.data, columns=iris.feature_names)
iris_df['species'] = iris.target
mapping = {0 : 'setosa', 1: 'versicolor', 2: 'virginica'}
iris_df = iris_df.replace({'species': mapping})
iris_df['sepal length (bins)'] = pd.cut(iris_df['sepal length (cm)'], bins=[0, 3, 6, 9], include_lowest=False, right=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment