熊猫:根据条件

时间:2018-06-19 03:53:30

标签: python pandas dataframe

我想在两个不同的列中填充两个数字之间的值。我有一个看起来像这样的数据框(df) 当我在开始列结束列中有一个数字时,我只想填充值列。 基本上,Start表示进程已开始,结束表示进程正在结束。因此,只能在这两者之间填写值列

这是生成数据的代码,DF是我想要的。

from datetime import datetime, timedelta
import pandas as pd
import numpy as np
import random
np.random.seed(11) 
date_today = datetime.now()
ndays = 10
df = pd.DataFrame({'date': [date_today + timedelta(days=x) for x in range(ndays)], 
               'Start': pd.Series(np.random.randn(ndays)),     'End':pd.Series(np.random.randn(ndays))})
 df = df.set_index('date')
 df = df.mask(np.random.random(df.shape) < .6)
 df.End[{0,1,2,5,6,9}]=np.nan
 df.Start[5]=1
 df.Start[{1,3,4,2,8, 9}]=np.nan
 df['Value']=np.nan
 df.Value[{0,5}]=[0.3,0.1]
 df

我想获得一个如下所示的数据帧(DF):

                                 End     Start  Value
date                                                 
2018-06-18 22:34:35.964286       NaN  1.749455    0.3
2018-06-19 22:34:35.964286       NaN       NaN    0.3
2018-06-20 22:34:35.964286       NaN       NaN    0.3
2018-06-21 22:34:35.964286  0.561192       NaN    0.3
2018-06-22 22:34:35.964286       NaN       NaN    NaN
2018-06-23 22:34:35.964286       NaN  1.000000    0.1
2018-06-24 22:34:35.964286       NaN       NaN    0.1
2018-06-25 22:34:35.964286       NaN       NaN    0.1
2018-06-26 22:34:35.964286 -0.031075       NaN    0.1
2018-06-27 22:34:35.964286       NaN       NaN    NaN

提前致谢

1 个答案:

答案 0 :(得分:2)

试试这个:首先你转发填充。然后计算'事件'的数量。如果“事件”的数量是偶数,则用NaN替换值。

df['Value'] = df['Value'].fillna(method='ffill')
temp = (df['End'].shift().notnull().astype(int) + df['Start'].notnull().astype(int)).cumsum()
df.loc[temp % 2 == 0, 'Value'] = np.nan

编辑:我们必须使用shift()来修改“结束”,否则它会计算它们出现的同一行上的“结束”事件。我们希望“结束”事件在它们发生后立即计算,以便“结束”行中的“值”无效。