通过在pandas数据框中也复制其他一些列来堆积价值？

debugcn 发表于 Dev

s_khan92

我有像这样的数据框：

ID              Brands          Age           Gender           City
1             BMW_Audi_VW       50             M               Milano
2               VW_BMW          45             F               SF

我想Brands在“ _”上拆分列，并希望复制所有列，除了City

我可以根据这样的ID列进行操作：

df = df.set_index('ID').stack().str.split('_', expand=True).unstack(-1).stack(0).reset_index()

但它仅重复ID列。我需要所有列，但不需要“城市”

这是我正在寻找的理想输出：

ID              Brands          Age           Gender           City
 1             BMW               50             M               Milano
 1             Audi              50             M               None
 1             VW                50             M               None
 2             VW                45             F               SF
 2             BMW               45             F               None

耶斯列尔

使用DataFrame.explode与分裂列中的值Series.str.split，然后设置NoneS按DataFrame.mask：

df = df.assign(Brands = df['Brands'].str.split('_')).explode('Brands')

include = ['ID','Brands','Age','Gender']
cols = df.columns.difference(include)
df[cols] = df[cols].mask(df.index.to_series().duplicated(), None)
df = df.reset_index(drop=True)
print (df)
   ID Brands  Age Gender    City
0   1    BMW   50      M  Milano
1   1   Audi   50      M    None
2   1     VW   50      M    None
3   2     VW   45      F      SF
4   2    BMW   45      F    None

编辑：

检查差异：

#Brands column is assigned to Brands column (to same column)
df1= df.assign(Brands = df['Brands '].str.split('_')).explode('Brands')
print (df1)
   ID Brands  Age Gender    City
0   1    BMW   50      M  Milano
0   1   Audi   50      M  Milano
0   1     VW   50      M  Milano
1   2     VW   45      F      SF
1   2    BMW   45      F      SF

#Brands column is assigned to Brands1 column (to another column)
df2 = df.assign(Brands1 = df['Brands'].str.split('_')).explode('Brands')
print (df2)
   ID       Brands  Age Gender    City          Brands1
0   1  BMW_Audi_VW   50      M  Milano  [BMW, Audi, VW]
1   2       VW_BMW   45      F      SF        [VW, BMW]

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。