內聯if語句條件pandas為新列賦值

Question

我正在嘗試使用pandas assign有條件地為列分配值。

我嘗試使用pandas assign來創建一個新列，如果列sv_length指定的長度值> = 50，則將其標記為SV，如果長度<50則使用InDel。

df3=df2.assign(InDel_SV='InDel' if df2.sv_length < 50 else 'SV')

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

其他示例使用np.where。 為什么我要使用numpy？ 這個簡單的功能不應該成為熊貓的一部分嗎？

https://chrisalbon.com/python/data_wrangling/pandas_create_column_using_conditional/

Answer 1

通過使用apply支持此語法。

df3 = df2.assign(
    InDel_SV=df2.sv_length.apply(lambda x: 'InDel' if x < 50 else 'SV'))

但是，為了提高性能，建議您使用numpy，因為apply是一個緩慢的便利功能 。 這樣做的pandaic方式是使用numpy.where ：

df3 = df2.assign(InDel_SV=np.where(df2.sv_length < 50, 'InDel', 'SV'))