代码之家  ›  专栏  ›  技术社区  ›  HHH

如何基于数据集中集合的大小创建列

  •  0
  • HHH  · 技术社区  · 7 年前

    A, B, C
    1, 2, {nan, v}
    1, 3, {nan, v,b}
    

    在这个例子中 A 和 B 是我在groupby中使用的列。我想在这个数据框中添加另一列,即列中的元素数 C

    A, B, C, D
    1, 2, {nan, v}, 1
    1, 3, {nan, v,b}, 2
    

    我该怎么做?

    1 回复  |  直到 7 年前
        1
  •  1
  •   BENY    7 年前

    使用

    df['D']=pd.DataFrame(df.C.tolist(),index=df.index).notnull().sum(axis=1)