Get all the keys from json reponse - Python

user_10726514 0
Viewed 61

My json data would look like this

{
   "a":1,
   "b":[
      {
         "c":2,
         "d":{
            "e":3
         },
         "f":{
            "g":4
         },
         "h":[
            {
               "i":5
            },
            {
               "j":6
            }
         ]
      }
   ]
}

Is there a way I can extract all the keys from the response. What I would want is like: [a,b,c,d,e,f,g,h,i,j]. Can this be done?

key api dictionary python json
4 Answers

You can solve it, using generators, for example:

def get_keys(d):
    if isinstance(d, dict):
        for k, v in d.items():
            yield k
            yield from list(get_keys(v))
    elif isinstance(d, list):
        for o in d:
            yield from list(get_keys(o))


print(list(get_keys(adict)))

Output:

['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

If you want pairs (key, value) see this:

def get_keys(d):
    if isinstance(d, dict):
        for k, v in d.items():
            if isinstance(v, (int, str)):
                yield (k, v)
            else:
                yield from list(get_keys(v))
    elif isinstance(d, list):
        for o in d:
            yield from list(get_keys(o))

print(dict(get_keys(adict)))

Output:

{'a': 1, 'c': 2, 'e': 3, 'g': 4, 'i': 5, 'j': 6}
user_99267210

Recursion (Depth-First Search) would be a go to for this:

def get_keys(data):
    keys = []
    if isinstance(data, dict):
        for key, value in data.items():
            keys.append(key)
            keys += get_keys(value)
    elif isinstance(data, list):
        for value in data:
            keys += get_keys(value)
    return keys

print(get_keys(data=response))
user_127833630

You can use recursive function and iterate over all items in dict.

def get_all_key(dct, lst):
    for k,v in dct.items():
        lst.append(k)
        if isinstance(v, list):
            for d in v:
                get_all_key(d, lst)
        elif isinstance(v, dict):
            get_all_key(v, lst)

res = []
get_all_key(dct, res)
print(res)

Output:

['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
user_17405770

Adding to the above correct answers, this is also a method.

I mean there are a lot of ways in which we can do this, but they say smallest amount of work is the best work.

If all you data only string as keys and no key in the string, we can do this:

import json
import re

json_text = json.dumps(a)
print([_.replace('"', "") for _ in re.findall(r"\"\w\"", json_text)])

Convert dictionary to json, find all the matching "a" where a is the character, then remove the " from all matches and voila..

['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
user_141861160
Related