python - 通过另一个数据框的值列表拆分 Pandas 数据框

Question

我很确定有一个非常简单的解决方案，但我只是没有意识到。然而……

我有一个高频数据的数据框。将此数据框称为 A。我还有一个单独的低频分界点列表，称为 B。如果 A 的时间戳列在 B[0] 和 B[1] 之间，我想向 A 添加一个显示 1 的列], 如果在B[1]和B[2]之间，则为2，依此类推。

如前所述，这可能微不足道，我只是在这么晚的时候才意识到这一点。

Answer 1

这是一种使用列表推导式的快速而肮脏的方法。

>>> df = pd.DataFrame({'A': np.arange(1, 3, 0.2)})

>>> A = df.A.values.tolist()
A: [1.0, 1.2, 1.4, 1.6, 1.8, 2.0, 2.2, 2.5, 2.6, 2.8]

>>> B = np.arange(0, 3, 1).tolist()
B: [0, 1, 2]

>>> BA = [k for k in range(0, len(B)-1) for a in A if (B[k]<=a) & (B[k+1]>a) or (a>max(B))]
BA: [0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]