python - Pandas:将单元格拆分成多列，写入NaN

Question

我有以下 df:

df = pd.DataFrame({'Category': ['root:catA', 'root:catA:catB'], 'Age':[32, 34]})

现在我想将 Category 列按 : 拆分为总共 4 列。由于第 0 行将产生 4 个填充列，第 1 行将产生 5 个填充列，因此列数将减少，其余列应填充 NaN。

预期输出:

df = pd.DataFrame({'Category': ['root:catA', 'root:catA:catB'], 'Age':[32, 34], 'Cat1':['root', 'root'], 'Cat2':['catA', 'catA'], 'Cat3':['NaN', 'CatB'], 'Cat4':['NaN', 'NaN']})

我尝试了以下方法:

df[['cat1','cat2', "cat3", "cat4"]] = df.Category.str.split(":",expand=True)

但是我得到了一个 ValueError:列的长度必须与键的长度相同

Answer 1

差不多了，您只需要一个额外的重建索引步骤:

df['Category'].str.split(':', expand=True).reindex(range(4), axis=1)                                                   

      0     1     2   3
0  root  catA  None NaN
1  root  catA  catB NaN

现在，分配将起作用。

或者，加入它们。

(df['Category'].str.split(':', expand=True)
               .reindex(range(4), axis=1)
               .rename(lambda x: f'cat{x+1}', axis=1)
               .join(df))

   cat1  cat2  cat3  cat4        Category  Age
0  root  catA  None   NaN       root:catA   32
1  root  catA  catB   NaN  root:catA:catB   34