Create another numpy.array in a pandas data based upon conditionals

Question

I have a dataframe df_a, with a numpy-array named 'Language'. I want to create another numpy-array, LanguageCode, based upon Language and the Language codes associated with a Language.

df_a = pd.DataFrame({'Language':[['cantonese', 'japanese', 
                 'mandarin','american'],['mandarin','english'], 
                 ['american', 'mandarin','cantonese']]})```

df_a

     Language                                  LangugeCode
0   [cantonese, japanese, mandarin, american]  [zh_yue,ja,cmn,us]
1   [mandarin, english]                        [cmn,en]
2   [american, mandarin, cantonese]            [us,cmn,zh_yue'

Where do you retrieve the language code? Using pycountry like this answer? — Bill Huang
– Bill Huang, Commented Oct 6, 2020 at 0:20

O Pardal · Accepted Answer · 2020-10-06 00:41:35Z

1

I assumed that you have a dictionary to associate language and language code, and then used map.

Please, check if it helps you:

Assumptions:

import pandas as pd
import numpy as np

df_a = pd.DataFrame({'Language':[['cantonese', 'japanese', 
                 'mandarin','american'],['mandarin','english'], 
                 ['american', 'mandarin','cantonese']]})

#this is the hypothetical dictionary
lang_codes = {'cantonese': 'zh_yue','japanese': 'ja', 'mandarin': 'cmn','american': 'us','english': 'en'}

What you can do:

df_a['Language Code'] = [list(map(lambda x: lang_codes[x], row)) for row in df_a.Language]

Checking:

#getting the numpy array format
language_code = np.array(df_a['Language Code'])

type(language_code)

numpy.ndarray

And your dataframe will be:

    Language                                    Language Code
0   [cantonese, japanese, mandarin, american]   [zh_yue, ja, cmn, us]
1   [mandarin, english]                         [cmn, en]
2   [american, mandarin, cantonese]             [us, cmn, zh_yue]

edited Oct 6, 2020 at 0:41

answered Oct 6, 2020 at 0:30

O Pardal

6825 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Jeff Magouirk Over a year ago

Yes, I have a list of matching code. I will try this. Thank you very much.

O Pardal Over a year ago

@JeffMagouirk, if this answer addresses your problem, please mark answer as accepted so community can read as a trusted solution. Thanks

Collectives™ on Stack Overflow

Create another numpy.array in a pandas data based upon conditionals

1 Answer 1

Assumptions:

What you can do:

Checking:

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Assumptions:

What you can do:

Checking:

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related