Skip to content

Instantly share code, notes, and snippets.

@npinto

npinto/labels_to_integers.py

Created September 4, 2012 21:36
Show Gist options
  • Save npinto/3626866 to your computer and use it in GitHub Desktop.
Save npinto/3626866 to your computer and use it in GitHub Desktop.
Download ZIP
String labels to sorted 0-based integers.
Raw
labels_to_integers.py
In [12]: l = ['a', 'a', 'c', 'c', 'c', 'b']
In [13]: u = np.unique(l)
In [14]: u
Out[14]:
array(['a', 'b', 'c'],
dtype='|S1')
In [15]: np.searchsorted(u, l)
Out[15]: array([0, 0, 2, 2, 2, 1])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment