根据 Pandas 中的条件将列值加一

Question

我创建了一个脚本，它会抓取 NBA 的比赛数据并将其组织成 pandas dataframe。 在每节结束和比赛结束时，“详细信息”列中显示以下值：

    Detail
End of the 1st Quarter  
End of the 2nd Quarter  
End of the 3rd Quarter  
End of the 4th Quarter  
End of Game 

有没有办法创建一个以数字 1 开头的“季度”列，然后使用“详细信息”列在每个季度结束后增加 1？ 例如：

Detail           Quarter
Shot by...          1
Rebound...          1
End of 1st Quarter  1
Pass to...          2

Answer 1

使用带有cumsum和 bfill 的bfill

import numpy as np
df['Quarter'] = np.where(
              df['Detail'].str.contains('Quarter'),
                             df['Detail'].str.contains('Quarter').cumsum(), 
                             np.nan
                       )

df['Quarter'] = df['Quarter'].bfill()

---

print(df)
   Detail               Quarter
0  Shot by...              1.0
1  Rebound...              1.0
2  End of 1st Quarter      1.0
3  Pass to...              2.0
4  Shot by...              2.0
5  Rebound...              2.0
6  End of 2nd Quarter      2.0