Create a function where you pass in a key to object in typescript

index.ts

type SubObjKey = keyof ISubObj;
type ObjKey = keyof IObj;

interface ISubObj {
  subKey1: number;
  subKey2: number;
}

interface IObj {
 key1: ISubObj;
 key2: ISubObj;
}

const myFunc = (data:IObj,key:ObjKey, subKey: SubObjKey) => data[key][subKey]