How to assign different numbers to Dataframe's one column

Question

I had a dataframe which has 294467 rows and 7 columns. I want to assign the same number to products' brands which has the same brand.

Here is the example of my dataframe:

        overall  ...                 brand
0           5.0  ...     Pirmal Healthcare
1           5.0  ...     Pirmal Healthcare
2           5.0  ...     Pirmal Healthcare
3           5.0  ...     Pirmal Healthcare
4           4.0  ...     Pirmal Healthcare
 ...        ...                   ...
294975      4.0  ...  Gentlemen's Hardware
294976      5.0  ...     Benefit Cosmetics
294977      1.0  ...         Salon Perfect
294978      1.0  ...               GBSTORE
294979      1.0  ...               GBSTORE

[294467 rows x 7 columns]

Final result should be:

        overall  ...    brand
0           5.0  ...     1
1           5.0  ...     1
2           5.0  ...     1
3           5.0  ...     1
4           4.0  ...     1
  ...       ...         ...
294975      4.0  ...    7839
294976      5.0  ...    7840
294977      1.0  ...    7841
294978      1.0  ...    7842
294979      1.0  ...    7842

[294467 rows x 7 columns]

For this result, I sorted my dataframe according to brand. Then assigned different numbers to them with this code:

sorted_copy = copy.sort_values('brand')

random_number=0
first=""
for f, row in sorted_copy.iterrows():  
    i=row['brand'] 
    
    if(first == i):
        sorted_copy.at[f, 'brand'] = random_number
        
        
    elif(first !=i):
        first=i
        random_number= random_number +1
        sorted_copy.at[f, 'brand'] = random_number

However, this process took maybe an hour and half. Is there any solution to get this result in a short time? Can anyone help?

Thank you.

Can you past a sample of your dataframe as well as expected output (it's not very clear exactly what you're trying to achieve) — Danail Petrov, Jan 10 '21 at 16:55
Could you check "brand" column of my examples please? The other columns are not that important. That's why I didn't add them here. @DanailPetrov — jazz kek, Jan 10 '21 at 16:58
So .. you just need to add a unique number for every brand, is that it? — Danail Petrov, Jan 10 '21 at 17:00
Yes, that's exactly what I want. I did it in my way but as I said it took too long time. :( @DanailPetrov — jazz kek, Jan 10 '21 at 17:01
This solution worked and it was really fast. Thank you! @anky — jazz kek, Jan 10 '21 at 17:13

score 0 · Answer 1 · answered Jan 10 '21 at 17:03

0

df['brand'] = df['brand'].astype("category").cat.codes should work fine.

answered Jan 10 '21 at 17:03

Utkarsh Prakash

13
1
3

How to assign different numbers to Dataframe's one column

1 Answers1