在另一列中基于上一个条件添加行

Question

我是熊猫舱的新手。还有一个关于数据操作的快速问题：

假设我有一张如下表：

Tool | WeekNumber | Status | Percentage
-----|------------|--------|------------
  M1 |     1      |   good |     85
  M1 |     4      |   bad  |     75
  M1 |     7      |   good |     90

根据状态的情况，我想增加百分比。在

例如：

1. 如果状态为“良好”，则后续周数的以下行应全部为100，即，下一行应为第2周和第3周的100%

2. 如果状态为“坏”，则接下来的周数的百分比应为0，即第5周和第6周的百分比应为0。

我对如何处理条件有一些想法，但不知道如何添加行：

^{pr2}$

但是，根据条件，这只给了我三个值，并更改了特定周数的值。但我要的是：

Tool | WeekNumber | Status | Percentage
-----|------------|--------|------------
  M1 |     1      |   good |     85
  M1 |     2      |   good |     100
  M1 |     3      |   good |     100
  M1 |     4      |   bad  |     75
  M1 |     5      |   bad  |      0
  M1 |     6      |   bad  |      0
  M1 |     7      |   good |     90

Answer 1

你的回答是这样的：

add_rows = []
for index, elem in enumerate(df.Status):
    if elem == "good":

        # assuming data is sorted by 'WeekNumber'
        add_rows.append({'Tool': 'M1', 'WeekNumber': index + 2}) # etc
        add_rows.append({'Tool': 'M1', 'WeekNumber': index + 3}) # etc

more_data = pd.DataFrame(add_rows)
df = pd.concat([df, more_data]).sort_values(by='WeekNumber')