Python Pandas: выберите данные и игнорируйте KeyErrors

Question

Примечание: мой вопрос не this one, но что-то немного более тонкое.

Скажем, у меня есть dataframe, который выглядит, как это

df = 
    A  B C 
0 3  3 1 
1 2  1 9 

df[["A", "B", "D"]] поднимет KeyError.

Есть ли способ python pandas позволить df[["A", "B", "D"]] == df[["A", "B"]]? (То есть: просто выберите столбцы, которые существуют.)

Одним из решений может быть

good_columns = list(set(df.columns).intersection(["A", "B", "D"])) 
mydf = df[good_columns] 

Но есть две проблемы:

• Это неуклюжим и безвкусный.
• Заказ mydf.columns может быть ["A", "B"] или ["B", "A"].

Answer 1

Вы можете использовать filter, это будет просто игнорировать любые дополнительные клавиши:

df.filter(["A","B","D"]) 
    A  B 
0 3  3 
1 2  1 

Answer 2

Вы можете использовать условный список понимание:

target_cols = ['A', 'B', 'D'] 
>>> df[[c for c in target_cols if c in df]] 
    A B 
0 3 3 
1 2 1