2017-10-19 1 views
2

Ich habe ein Datenframe mit Spalten, die "pre_1, pre_2, pre_3, etc ..." enthalten. Ich habe auch Post-Daten als „post1 gekennzeichnet, post2, Post3 ... etc).Subsetting Dataframe nach Spalte Name

Gibt es eine Möglichkeit, Daten in Python zu schneiden durch alle Spalten mit dem Wort 'pre' Auswahl?

Dank.

Antwort

0

Sie können filter mit like oder regex Schlüsselwort Muster in den Spaltennamen übereinstimmen:

df = pd.DataFrame({ 
    'pre_1': [1,2], 
    'pre_2': [3,4], 
    'pre_3': [5,6], 
    'post1': [7,8], 
    'post2': [9,10], 
    'post3': [11,12] 
}) 

df 
#post1 post2 post3 pre_1 pre_2 pre_3 
#0 7  9  11  1  3  5 
#1 8  10  12  2  4  6 

So wählen Sie Spalten th in ein Muster enthalten, verwenden Sie das like Stichwort:

df.filter(like='pre') 

#pre_1 pre_2 pre_3 
#0 1  3  5 
#1 2  4  6 

Oder können Sie reguläre Ausdrücke verwenden, werden die folgenden Spalten auswählen, die beginnt (^) mit pre:

df.filter(regex='^pre') 

#pre_1 pre_2 pre_3 
#0 1  3  5 
#1 2  4  6 
Verwandte Themen