Get all the keys from json reponse – Python

Question

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]. Could this be done?

Asked By: Joxin Jose

||

Source

Answer 1

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}

Answered By: funnydman

Answer 2

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))

Answered By: Jobo Fernandez

Answer 3

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']

Answered By: I'mahdi

Answer 4

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']

Answered By: Nitish

Get all the keys from json reponse – Python

Question:

Answers: