Python: Getting value in dictionary by key and creating an array with them

Question

I have a dictionary like this:

{
  "0": {}, 
  "1": {}, 
  "2": {}, 
  "3": {}, 
  "4": {}, 
  "5": {}, 
  "6": {}, 
  "7": {}, 
  "8": {}, 
  "9": {}, 
  "10": {}, 
  "11": {}, 
  "12": {
    "PNR": "51313542"
  }, 
  "13": {}, 
  "14": {}, 
  "15": {
    "PNR": "13151351351"
  }, 
  "16": {
    "PNR": "13151351351"
  }, 
  "17": {
    "PNR": "13151351351"
  }, 
  "18": {
    "PNR": "13151351351"
  }, 
  "19": {
    "PNR": "13151351351"
  }, 
  "20": {
    "PNR": "13151351351"
  }, 
  "21": {
    "PNR": "13151351351"
  }, 
  "22": {
    "PNR": "13151351351"
  }, 
  "23": {
    "PNR": "13151351351"
  }, 
  "24": {
    "PNR": "13151351351"
  }
}

I wanted to get only the PNR's and creating an array with them.

I tried this for getting the PNR values (think "output" as my dict):

d_ = output.get("PNR")

But it gives None.

How can I get the PNR values no matter how many elements in my dict and create an array with them?

What should happen if there is no PNR key in specific sub-dict? — buran
– buran, Commented Nov 15, 2021 at 20:51

Tranbi · Accepted Answer · 2021-11-15 20:53:18Z

3

You could use a list comprehension:

[v["PNR"] for v in dic.values() if "PNR" in v.keys()]

Output:

['51313542', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351']

answered Nov 15, 2021 at 20:53

Tranbi

12.8k6 gold badges19 silver badges39 bronze badges

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

Comments

cards · Accepted Answer · 2021-11-15 20:54:52Z

2

In case the PNRs are not all at the same level you can use recursive approach:

def get_pnr(dct):
    pnrs = []
    for k, v in dct.items():
        if k == 'PNR':
            pnrs += [v]
        if isinstance(v, dict):
            pnrs.extend(get_pnr(v))
            
    return pnrs

a = get_pnr(d)
print(a)

Output

['51313542', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351', '13151351351']

answered Nov 15, 2021 at 20:54

cards

5,2801 gold badge13 silver badges28 bronze badges

1 Comment

juanpa.arrivillaga Over a year ago

pnrs += [v] should be pnrs.append(v)...

Collectives™ on Stack Overflow

Python: Getting value in dictionary by key and creating an array with them

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related