Berikut ini lebih lambat daripada pendekatan waktu di sini , tetapi kami dapat menghitung kolom tambahan berdasarkan konten lebih dari satu kolom, dan lebih dari dua nilai dapat dihitung untuk kolom tambahan.
Contoh sederhana hanya menggunakan kolom "Set":
def set_color(row):
if row["Set"] == "Z":
return "red"
else:
return "green"
df = df.assign(color=df.apply(set_color, axis=1))
print(df)
Set Type color
0 Z A red
1 Z B red
2 X B green
3 Y C green
Contoh dengan lebih banyak warna dan lebih banyak kolom yang diperhitungkan:
def set_color(row):
if row["Set"] == "Z":
return "red"
elif row["Type"] == "C":
return "blue"
else:
return "green"
df = df.assign(color=df.apply(set_color, axis=1))
print(df)
Set Type color
0 Z A red
1 Z B red
2 X B green
3 Y C blue
Sunting (21/06/2019): Menggunakan plydata
Dimungkinkan juga untuk menggunakan plydata untuk melakukan hal-hal semacam ini (ini tampaknya lebih lambat daripada menggunakan assign
dan apply
, meskipun).
from plydata import define, if_else
Sederhana if_else
:
df = define(df, color=if_else('Set=="Z"', '"red"', '"green"'))
print(df)
Set Type color
0 Z A red
1 Z B red
2 X B green
3 Y C green
Bersarang if_else
:
df = define(df, color=if_else(
'Set=="Z"',
'"red"',
if_else('Type=="C"', '"green"', '"blue"')))
print(df)
Set Type color
0 Z A red
1 Z B red
2 X B blue
3 Y C green