如何在 pandas dataframe 中添加重復值的列？

Question

如果我有一個 pandas dataframe 看起來像這樣：

     X [ m ]  Y [ m ]  Z [ m ]
0         1     1      0.0
1         2     0.5    0.1
2         3     2      0.3
3         4     1      0.4
4         5     3      0.5 
5         1     4      0.6
6         2     1.5    0.8
7         3     6      1.0
8         4     3      1.2
9         5     4      1.5
...

我將如何添加看起來像這樣的第四列 (Z_2)（僅包括 0.5 跳躍的值）：

     X [ m ]  Y [ m ]  Z [ m ]  Z_2 [ m ]
0         1     1      0.0      0.0
1         2     0.5    0.1      0.0
2         3     2      0.3      0.5
3         4     1      0.4      0.5
4         5     3      0.5      0.5
5         1     4      0.6      0.5
6         2     1.5    0.8      1.0 
7         3     6      1.0      1.0
8         4     3      1.2      1.0
9         5     4      1.5      1.5 
...

Answer 1

你可以試試：

df["Z_2"] = (df["Z"] // 0.5) * 0.5

如果你在談論 Z 列中只有 0.5 個“跳躍”。

Answer 2

你可以嘗試這樣的事情：

df["Z_2"] = (df["Z"] % .5 == 0).cumsum() * .5

這里df["Z"] %.5 == 0找到df["Z"]中位於 .5 邊界上的值， cumsum只會在遇到True時加 1。