Assign a number to each unique value in a list

by

Without using an external library (check the EDIT for a Pandas solution) you can do it as follows :

d = {ni: indi for indi, ni in enumerate(set(names))}
numbers = [d[ni] for ni in names]

Brief explanation:

In the first line, you assign a number to each unique element in your list (stored in the dictionary d; you can easily create it using a dictionary comprehension; set returns the unique elements of names).

Then, in the second line, you do a list comprehension and store the actual numbers in the list numbers.

One example to illustrate that it also works fine for unsorted lists:

# 'll' appears all over the place
names = ['ll', 'll', 'hl', 'hl', 'hl', 'LL', 'LL', 'll', 'LL', 'HL', 'HL', 'HL', 'll']

That is the output for numbers:

[1, 1, 3, 3, 3, 2, 2, 1, 2, 0, 0, 0, 1]

As you can see, the number 1 associated with ll appears at the correct places.

EDIT

If you have Pandas available, you can also use pandas.factorize (which seems to be quite efficient for huge lists and also works fine for lists of tuples as explained here):

import pandas as pd

pd.factorize(names)

will then return

(array([(array([0, 0, 1, 1, 1, 2, 2, 0, 2, 3, 3, 3, 0]),
 array(['ll', 'hl', 'LL', 'HL'], dtype=object))

Therefore,

numbers = pd.factorize(names)[0]

More Related Contents:

  1. How to iterate through two lists in parallel?
  2. Transpose/Unzip Function (inverse of zip)?
  3. Are lists thread-safe?
  4. Python List vs. Array – when to use?
  5. How can I iterate through two lists in parallel?
  6. Getting a list of values from a list of dicts
  7. Find and replace string values in list
  8. Pairs from single list
  9. Splitting a list based on a delimiter word
  10. How do you pick “x” number of unique numbers from a list in Python?
  11. Naming conflict with built-in function
  12. How do I split a list into equally-sized chunks?
  13. How can I make a dictionary from separate lists of keys and values?
  14. What is the time complexity of popping elements from list in Python?
  15. How can I generate a list of consecutive numbers?
  16. Accessing a value in a tuple that is in a list
  17. Python split string into multiple string [duplicate]
  18. Python passing list as argument
  19. Python Bool and int comparison and indexing on list with boolean values
  20. Best way to check if an item is present in a list of lists? [duplicate]
  21. How can I use list comprehensions to process a nested list?
  22. Why does reversing a list using slice notation with 0 as “stop” not return the entire list?
  23. How to access a column in a list of lists in python
  24. Loop over 2 lists, repeating the shortest until end of longest [duplicate]
  25. Duplicate elements in a list [duplicate]
  26. Python: iterating over list vs over dict items efficiency
  27. First items in inner list efficiently as possible [duplicate]
  28. Python finding repeating sequence in list of integers?
  29. Find non-common elements in lists
  30. Automatically growing lists in Python

Leave a Comment