根據兩個條件選擇python數據框中的行

Question

基於下面的數據框（1），我希望創建一個數據框（2），其中y或z等於2.是否有方法可以方便地執行此操作？

如果我要創建一個僅包含dataframe（1）但不包含dataframe（2）的行的數據框（3），我該如何處理它？

id x y z 
    0 324 1 2 
    1 213 1 1 
    2 529 2 1 
    3 347 3 2 
    4 109 2 2 

...

Answer 1

您可以輕鬆創建df2足夠使用條件：

df2 = df1[df1.y.eq(2) | df1.z.eq(2)] 

df2 
     x y z 
id   
0 324 1 2 
2 529 2 1 
3 347 3 2 
4 109 2 2 

鑑於df2和df1，你可以對指數進行組差異操作，如此：

df3 = df1.iloc[df1.index.difference(df2.index)] 

df3 
     x y z 
id   
1 213 1 1 

Answer 2

你可以做到以下幾點：

import pandas as pd 

df = pd.read_csv('data.csv') 
df2 = df[(df.y == 2) | (df.z == 2)] 

print(df2) 

結果：

id x y z 
0 0 324 1 2 
2 2 529 2 1 
3 3 347 3 2 
4 4 109 2 2 

Answer 3

df[df[['y','z']].eq(2).any(1)] 
Out[1205]: 
    id x y z 
0 0 324 1 2 
2 2 529 2 1 
3 3 347 3 2 
4 4 109 2 2