npinto/labels_to_integers.py

Created September 4, 2012 21:36

Star (1) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/npinto/3626866.js"></script>
Save npinto/3626866 to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

npinto created this gist Sep 4, 2012.

labels_to_integers.py

-Original file line number
+Diff line change
@@ -0,0 +1,11 @@
+    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])