python - 当特定列达到给定值时分割 Pandas 框架

Question

我想为数据帧的两个特定部分使用两种不同的 bin 大小的 df.cut() 。我相信最简单的方法是读取我的数据帧并将其分成两部分，这样我就可以在具有两个独立容器的两个独立数据帧中使用 df.cut() 。

我知道我可以使用 df.head()，但我必须不断更改数据帧，而且它们的大小并不总是相同。例如，使用以下数据框

               A     B
          1    0.1   0.423655
          2    0.2   0.645894
          3    0.3   0.437587
          4    0.31  0.891773
          5    0.4   0.1773
          6    0.43  0.91773
          7    0.5   0.891773

我想要两个数据帧，A 的值高于或等于 0.4 且低于 0.4。

所以我会有 df2:

                 A     B
          1    0.1   0.423655
          2    0.2   0.645894
          3    0.3   0.437587
          4    0.31  0.891773

和 df3:

                 A      B
           1    0.4   0.1773
           2    0.43  0.91773
           3    0.5   0.891773

同样，df.head(4) 或 df.tail(3) 不起作用。

Answer 1

df2 = df[df["A"] < 0.4]
df3 = df[df["A"] >= 0.4]