pandas 多条件过滤 修改列值

Question

import pandas as pd

data = [
    {'name': 'APPLE', 'price':6000,'stock': 1000, 'static':'A'},
    {'name': 'HUAWEI', 'price':5000,'stock': 510,'static':'B'},
    {'name': 'XIAOMI', 'price':4000,'stock': 300,'static':'B'},
     {'name': 'XIAOLJ', 'price':3000,'stock': 470,'static':'A'},
    {'name': 'Other', 'price':2999,'stock': 1200,'static':'A'},
]

df = pd.DataFrame(data)

条件
stock 小于 等于 600
price 大于 3000
结果
把 static 改为 A

其他不变，这个要怎么做？

不知道多个条件应该怎么写？apply

Answer 1

用了一个比较笨的方法，拿出符合条件的行的索引值，存入一个list，之后逐行修改。

import pandas as pd

data = [
    {'name': 'APPLE', 'price':6000,'stock': 1000, 'static':'A'},
    {'name': 'HUAWEI', 'price':5000,'stock': 510,'static':'B'},
    {'name': 'XIAOMI', 'price':4000,'stock': 300,'static':'B'},
     {'name': 'XIAOLJ', 'price':3000,'stock': 470,'static':'A'},
    {'name': 'Other', 'price':2999,'stock': 1200,'static':'A'},
]

df = pd.DataFrame(data)


list = df[(df["stock"] <= 600) & (df["price"] > 3000)].index
for i in list:
    df.loc[i,["static"]] = ['A']
    
print(df)

希望能帮到你。