如何在JavaScript中正确扩展Object类并保留类型提示？

你遇到的核心问题是TypeScript无法识别Object.assign动态添加的属性，导致实例丢失传入对象的类型提示。以下是两种可行的解决方案：

方案一：泛型类实现 #

通过给Dataframe类添加泛型参数，捕获传入对象的具体类型，让TypeScript自动推导实例的完整类型（包含传入对象的属性+类自身的属性/方法）：

class Dataframe<T extends Record<string, any[]>> {
  _nrow: number;
  _ncol: number;
  _identity: number[];
  // 添加索引签名，兼容动态添加的属性类型
  [key: string]: any[] | number | number[];

  constructor(values: T) {
    const keys = Object.keys(values);
    this._nrow = values[keys[0]].length;
    this._ncol = keys.length;
    this._identity = Array(this._nrow).fill(1);
    Object.assign(this, values);
  }

  // 示例方法：计算指定列的总和
  sum(col: keyof T) {
    return this[col].reduce((acc, val) => acc + val, 0);
  }
}

使用示例：

const rawData = {x : [1, 2, 3, 4], y: [5, 6, 7, 8]}   
const data = new Dataframe(rawData);

data.x; // 类型提示为number[]，无报错
data._nrow; // 正常访问类自身属性
data.sum('y'); // 方法参数自动提示x/y可选值

方案二：工厂函数实现 #

如果不需要类的继承特性，用工厂函数结合交叉类型的方式更简洁，类型提示也更直接：

// 定义Dataframe的基础属性和方法类型
type DataframeBase = {
  _nrow: number;
  _ncol: number;
  _identity: number[];
  sum(col: string): number;
};

function createDataframe<T extends Record<string, any[]>>(values: T): T & DataframeBase {
  const keys = Object.keys(values);
  const base: DataframeBase = {
    _nrow: values[keys[0]].length,
    _ncol: keys.length,
    _identity: Array(values[keys[0]].length).fill(1),
    sum(col) {
      return values[col].reduce((acc, val) => acc + val, 0);
    }
  };
  // 合并传入对象和基础属性方法
  return Object.assign({}, values, base);
}

使用示例：

const rawData = {x : [1, 2, 3, 4], y: [5, 6, 7, 8]}   
const data = createDataframe(rawData);

data.x; // 正常类型提示
data.sum('x'); // 方法调用正常

两种方案的核心都是利用TypeScript泛型捕获传入对象的具体类型，让类型系统能识别动态添加的属性，同时保留类/工厂函数提供的额外功能。

内容的提问来源于stack exchange，提问作者Adam B.