当对应的值为NaN时，如何根据另一列中的条件从前一行获取一个值？

Question

我正在使用时间序列数据框。 我正在寻找一种方法来找出 col2 的值，以及当 col1 的值为 1 时的 col3。所以，我正在寻找的输出是 col2：7 和 col3：11。我给出了下面的代码以供参考。 提前致谢！

d = {'col1': [0,0,1,0,0], 'col2' : [0,7,'N/A',9,10], 'col3': [11,'N/A','N/A',14,15]}
index = pd.DatetimeIndex(['2014-07-04', '2014-08-04', '2015-07-04', '2015-08-04', '2015-09-04'])
d = pd.DataFrame(data = d, index = index)
d = d.replace('N/A', np.nan)

Answer 1

您可以ffill值并以 1 作为值获取第一行的索引：

d.ffill().loc[d['col1'].eq(1).idxmax(), ['col2', 'col3']]

输出：

col2     7.0
col3    11.0
Name: 2015-07-04 00:00:00, dtype: float64

如果您期望有几行值为 1：

d.ffill().loc[d['col1'].eq(1), ['col2', 'col3']]

输出：

            col2  col3
2015-07-04   7.0  11.0