DataFrame删除指定值的行

182 阅读1分钟

这行代码的作用是删除 data_2 DataFrame 中 discountePrice 列值为 1 的行。用一个简单的例子来说明这个应用:

假设有一个名为 data_2 的 DataFrame,包含了产品的信息,其中一列是 discountePrice 表示折扣价格。我们想要删除折扣价格为 1 的产品行。

import pandas as pd

# 创建示例 DataFrame
data_2 = pd.DataFrame({
    'product': ['A', 'B', 'C', 'D'],
    'discountePrice': [0, 1, 1, 2]
})

print("Original DataFrame:")
print(data_2)

# 删除折扣价格为 1 的行
data_2 = data_2.drop(index=data_2[data_2.discountePrice == 1].index.tolist())

print("\nDataFrame after removing rows with discountePrice == 1:")
print(data_2)

在这个例子中,原始的 data_2 DataFrame 包含了四个产品的信息,其中两个产品的折扣价格为 1。然后,使用 drop() 函数结合条件选择来删除折扣价格为 1 的行。最后,打印输出删除后的 DataFrame。

输出如下:

Original DataFrame:
  product  discountePrice
0       A               0
1       B               1
2       C               1
3       D               2

DataFrame after removing rows with discountePrice == 1:
  product  discountePrice
0       A               0
3       D               2

可以看到,折扣价格为 1 的产品行已经被成功删除了,输出中只剩下了折扣价格不为 1 的产品行。