Pandas 根据特殊要求拆分一栏

Question

我有一个示例数据，如下所示。开始和结束在列中配对。

而且由于实际数据很大，所以不知道一个Start和End之间有多少行。

df = pd.DataFrame({'Item':['Item_A','<Start>','A1','A2','<End>','Item_B','<Start>','B1','B2','B3','<End>']})

print (df)
       Item
0    Item_A
1   <Start>
2        A1
3        A2
4     <End>
5    Item_B
6   <Start>
7        B1
8        B2
9        B3
10    <End>

如何用Pandas改成下面的格式？谢谢

Answer 1

如果 Item 值比 Start 行高出一行，解决方案有效：

#compare for Start
m1 = df['Item'].eq('<Start>')
#get Item rows by shift Start mask
m2 = m1.shift(-1, fill_value=False)
#replace non Item values to missing values and forward filling them
df['new'] = df['Item'].where(m2).ffill()
#compare End
m3 = df['Item'].eq('<End>')

#filter no matched rows between Start and End to m4
g = m1.cumsum()
s = m3.shift().cumsum()
m4 = s.groupby(g).transform('min').ne(s)

#filtering with swap columns
df1 = df.loc[~(m4 | m1 | m3), ['new','Item']].copy()
#new columns names
df1.columns = ['Item','Detail']
#replace duplicated to empty string
df1['Item'] = np.where(df1['Item'].duplicated(), '', df1['Item'])
print (df1)
     Item Detail
2  Item_A     A1
3             A2
7  Item_B     B1
8             B2
9             B3

Pandas 根据特殊要求拆分一栏

Pandas split one column according to a special requirement

python

rows

pandas