計算，如果價值在大熊貓數據幀列表上的一列

Question

我有我的數據幀2列存在由客戶「P」

產品ID列表購買

1. 產品ID通過類似的客戶購買「 P_LIST」

   df = pd.DataFrame({'p': [12, 4, 5, 6, 7, 7, 6,5],'p_list':[[12,1,5], [3,1],[8,9,11], [6,7,9], [7,1,2],[12,9,8], [6,1,15],[6,8,9,11]]}) 
   

我要檢查，如果 「p P_LIST」 上存在的 「」 還是不行，所以我申請這個代碼

df["exist"]= df.apply(lambda r: 1 if r["p"] in r["p_list"] else 0, axis=1) 

問題是我在這個數據幀中有大約5000萬行，所以執行需要很長時間。

是否有更有效的方法來計算此列？

謝謝。

Answer 1

您可以使用list comprehension，上投True, False值int：

df["exist"] = [r[0] in r[1] for r in zip(df["p"], df["p_list"])] 
df["exist"] = df["exist"].astype(int) 
print (df) 
    p   p_list exist 
0 12  [12, 1, 5]  1 
1 4   [3, 1]  0 
2 5  [8, 9, 11]  0 
3 6  [6, 7, 9]  1 
4 7  [7, 1, 2]  1 
5 7  [12, 9, 8]  0 
6 6  [6, 1, 15]  1 
7 5 [6, 8, 9, 11]  0 

---

df["exist"] = [int(r[0] in r[1]) for r in zip(df["p"], df["p_list"])] 
print (df) 
    p   p_list exist 
0 12  [12, 1, 5]  1 
1 4   [3, 1]  0 
2 5  [8, 9, 11]  0 
3 6  [6, 7, 9]  1 
4 7  [7, 1, 2]  1 
5 7  [12, 9, 8]  0 
6 6  [6, 1, 15]  1 
7 5 [6, 8, 9, 11]  0 

時序：

#[8000 rows x 2 columns] 
df = pd.concat([df]*1000).reset_index(drop=True) 
print (df) 

In [89]: %%timeit 
    ...: df["exist2"] = [r[0] in r[1] for r in zip(df["p"], df["p_list"])] 
    ...: df["exist2"] = df["exist2"].astype(int) 
    ...: 
100 loops, best of 3: 6.07 ms per loop 

In [90]: %%timeit 
    ...: df["exist"] = [1 if r[0] in r[1] else 0 for r in zip(df["p"], df["p_list"])] 
    ...: 
100 loops, best of 3: 7.16 ms per loop 

In [91]: %%timeit 
    ...: df["exist"] = [int(r[0] in r[1]) for r in zip(df["p"], df["p_list"])] 
    ...: 
100 loops, best of 3: 9.23 ms per loop 

In [92]: %%timeit 
    ...: df['exist1'] = df.apply(lambda x: x.p in x.p_list, axis=1).astype(int) 
    ...: 
1 loop, best of 3: 370 ms per loop 

In [93]: %%timeit 
    ...: df["exist"]= df.apply(lambda r: 1 if r["p"] in r["p_list"] else 0, axis=1) 
1 loop, best of 3: 310 ms per loop