我有這個(gè)Excel函數(shù)，它說(shuō)：=IF(id ="pre_stage";"a";IF(id="static";"b";"c"))我嘗試在我的 python 腳本中實(shí)現(xiàn)這一點(diǎn)來(lái)創(chuàng)建一個(gè)新列。df['type'] = df['id'].apply(lambda x: 'a' if x == 'pre_stage' else 'b')我錯(cuò)過(guò)了第二個(gè)條件，如果“id”是“static”，那么類型應(yīng)該是“b”，其他將是“c”。劇本應(yīng)該怎么寫？謝謝
3 回答
犯罪嫌疑人X
TA貢獻(xiàn)2080條經(jīng)驗(yàn) 獲得超4個(gè)贊
您可以傳入一個(gè)函數(shù)來(lái)代替使用 lambda apply：
def f(x):
if x == 'pre_stage':
return 'a'
elif x == 'static':
return 'b'
return 'c'
df['type'] = df['id'].apply(f)
您還可以使用字典：
d = {'pre_stage': 'a', 'static': 'b'}
df['type'] = df['id'].apply(lambda x: d.get(x, 'c'))
米琪卡哇伊
TA貢獻(xiàn)1998條經(jīng)驗(yàn) 獲得超6個(gè)贊
您可以在此處創(chuàng)建映射并使用pd.Series.map。
mapping?=?{"pre_stage":?"a",?"static":?"b"}
df["type"]?=?df["id"].map(mapping).fillna("c")你可以np.select在這里使用。
condlist?=?[df["id"].eq("pre_stage"),?df["id"].eq("static")]
choicelist?=?["a",?"b"]
df["type"]?=?np.select(condlist,?choicelist,?"c")
catspeake
TA貢獻(xiàn)1111條經(jīng)驗(yàn) 獲得超0個(gè)贊
如果您的條件要嵌套，最好為其定義一個(gè)函數(shù)。它還有助于使您的代碼更清晰。
import pandas as pd
df = pd.DataFrame({'type':['pre-stage','static','not-related']})
def func(x):
if x == 'pre-stage':
return 'a'
elif x == 'static':
return 'b'
return 'c'
結(jié)果：
df['type'].apply(func) #as example, you can assign it to the frame as you did
>>
0 a
1 b
2 c
Name: type, dtype: object
添加回答
舉報(bào)
0/150
提交
取消