python - 如何在python中剪切一个pandas列，并使另一列的长度相等？

Question

我有一个由几列组成的数据框，其中有 X、Y 和 Z 坐标。现在这些列的长度都相等。如果你想象一个圆柱体，我想做的就是切割圆柱体的一部分，使其变得更短。我已经设法弄清楚如何根据 Z 列中的实际值来剪切 Z 列: cutted_z = [i for i in df["Z"] if i >= 0 and i <= 1000]

如何将 X Y Z 相对于 Z 中的实际数值进行等分？

我当前解决方案的问题在于，它保留了 X 和 Y 列的长度以及数据框中的其余列，未受影响，这意味着 X 和 Y 现在在 Z 中有更多列。

Answer 1

你可以这样做:

mask = (df['Z'] >= 0) & (df['Z'] <= 1000) #creates a mask to filter dataframe on

df = df[mask]

这将为您提供所需 Z 值范围的数据帧切片。