如何在有字段值的角度返回嵌套对象数组的特定字段？

// 嵌套对象数组
const data = [
  {
    id: 1,
    name: 'John',
    address: {
      city: 'New York',
      country: 'USA'
    }
  },
  {
    id: 2,
    name: 'Jane',
    address: {
      city: 'London',
      country: 'UK'
    }
  }
];

// 返回嵌套对象数组的特定字段
function getSpecificFields(data, fields) {
  const result = [];
  
  for (let i = 0; i < data.length; i++) {
    const obj = {};
    
    for (let j = 0; j < fields.length; j++) {
      const field = fields[j];
      
      if (field.includes('.')) {
        const nestedFields = field.split('.');
        let value = data[i];
        
        for (let k = 0; k < nestedFields.length; k++) {
          value = value[nestedFields[k]];
        }
        
        obj[field] = value;
      } else {
        obj[field] = data[i][field];
      }
    }
    
    result.push(obj);
  }
  
  return result;
}

// 调用函数并指定需要返回的字段
const fieldsToReturn = ['id', 'name', 'address.city'];
const result = getSpecificFields(data, fieldsToReturn);

console.log(result);

[
  {
    id: 1,
    name: 'John',
    'address.city': 'New York'
  },
  {
    id: 2,
    name: 'Jane',
    'address.city': 'London'
  }
]