使用特定参数搜索数据框并存储相关值

Question

我有一个包含以下信息的数据框。

STATE         VOLTS
Discharge     2.31725         
Discharge     2.31714         
Rest          2.34857        
Rest          2.38084         
Rest          2.38182        
Rest          2.38222        
Rest          2.38651        
Rest          2.38643        
Rest          2.38647        
Rest          2.38651        
Rest          2.38647        
Discharge     2.32013       
Discharge     2.31926        

我想做的是让我的脚本通过“状态”列，如果它看到从“放电”到“休息”的变化，它应该执行数学运算：2.34857 - 2.31714 = 0.03143 并存储该值在新的数据框中。

但是，当向下进入“状态”列时，如果它看到“休息”到“放电”，它应该只存储与“休息”值关联的“伏特”。在这种情况下，2.38647。理想情况下，该值也应与前一个值存储在同一数据框中。

Answer 1

定义一个条件，当为真时，将值作为名为@9​​87654321@ 的新列返回：

row_cond = (df.STATE == "Rest") & (df.STATE.shift(1) == "Discharge")
new_col = "value"
df.loc[row_cond, new_col] = 2.34857 - 2.31714