Python Dictionary comprehension with condition

Question

Suppose that I have a dict named data like below:

{
  001: {
    'data': {
      'fruit': 'apple',
      'vegetable': 'spinach'
    },
    'text': 'lorem ipsum',
    'status': 10
  },
  002: {
    .
    .
    .
  }
}

I want to flatten(?) the data key and convert it to this:

{
  001: {
    'fruit': 'apple',
    'vegetable': 'spinach',
    'text': 'lorem ipsum',
    'status': 10
  },
  002: {
    .
    .
    .
  }
}

I am trying to achieve this using dict comprehensions. Below implementation is with for loops:

mydict = {}
for id, values in data.items():
    mydict[id] = {}
    for label, value in values.items():
        if label == 'data':
            for x, y in value.items():
                mydict[id][x] = y
        else:
            mydict[id][label] = value

I tried below comprehension but it gives syntax error:

mydict = {
    id: {x: y} for x, y in value.items() if label == 'data' else {label: value}
    for id, values in data.items() for label, value in values.items()}

Is there a way to achieve this using comprehensions only?

Answer 1

With dict expansions:

mydict = {i:{**v['data'], **{k:u for k, u in v.items() if k != "data"}} for i, v in data.items()}