簡體   English   中英

Pandas DataFrame中的新列,基於列表中是否有任何值出現在數據集中

[英]New column in Pandas DataFrame based on whether any value from a list appears in the dataset

我試圖根據值列表是否出現在數據框中向數據框中添加新列。 這是一個玩具示例:

我有一個看起來像這樣的數據框:

train = pd.DataFrame({'a' : [22, 45, 68],'b' : [11, 23, 89], 'c' : [17, 49, 56]})

    a   b   c
0   22  11  17
1   45  23  49
2   68  89  56

以及值列表:

a=[17,21,12,89,77,99]

我想根據我的值列表“ a”中是否有任何值出現在數據集中添加一列。 為此,我可以執行以下操作:

train["is_in_a"] = (train["a"].isin(a) | train["b"].isin(a) | train["c"].isin(a)).astype("int")

    a   b   c   is_in_a
0   22  11  17     1
1   45  23  49     0
2   68  89  56     1

對於小型數據集,這是可以的。 但是,我想對具有數千列的數據集執行此操作,因此對於如此大的情況,我無法編寫此語句。 有沒有一種相同但更有效,更整齊的書寫方式? 謝謝!!

我相信簡單的apply就能做到

train.apply(lambda k: any(x in a for x in k), axis = 1)

要么

train.isin(a).apply(any, 1)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM