Panda's: kolommen verwijderen die niet in de lijst staan

U kunt de volgende basissyntaxis gebruiken om kolommen uit een Panda DataFrame te verwijderen die niet in een specifieke lijst staan:

 #define columns to keep
keep_cols = [' col1 ', ' col2 ', ' col3 ']

#create new dataframe by dropping columns not in list
new_df = df[df. columns . intersection (keep_cols)]

In dit specifieke voorbeeld worden alle kolommen uit het DataFrame verwijderd die niet gelijk zijn aan col1 , col2 of col3 .

Het volgende voorbeeld laat zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld: verwijder kolommen die niet in de lijst staan in Pandas

Stel dat we het volgende panda’s DataFrame hebben dat informatie bevat over verschillende basketbalspelers:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12],
                   ' steals ': [4, 4, 10, 12, 8, 5, 5, 2]})

#view DataFrame
print (df)

  team points assists rebounds steals
0 A 18 5 11 4
1 B 22 7 8 4
2 C 19 7 10 10
3 D 14 9 6 12
4 E 14 12 6 8
5 F 11 9 5 5
6 G 20 9 9 5
7:28 4 12 2

Stel nu dat we een nieuw DataFrame willen maken dat alle kolommen verwijdert die niet in de volgende kolommenlijst voorkomen: team , points en Steles .

We kunnen hiervoor de volgende syntaxis gebruiken:

 #define columns to keep
keep_cols = [' team ', ' points ', ' steals ']

#create new dataframe by dropping columns not in list
new_df = df[df. columns . intersection (keep_cols)]

#view new dataframe
print (new_df)

  team points steals
0 to 18 4
1 B 22 4
2 C 19 10
3 D 14 12
4 E 14 8
5 F 11 5
6 G 20 5
7:28 a.m. 2

Houd er rekening mee dat alle kolommen in het oorspronkelijke DataFrame die niet in de lijst keep_cols staan, zijn verwijderd uit het nieuwe DataFrame.

Aanvullende bronnen

In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende panda-taken kunt uitvoeren:

Hoe de eerste rij bij panda’s te verwijderen
Hoe de eerste kolom in Panda’s te verwijderen
Hoe dubbele kolommen in Pandas te verwijderen
Hoe alle kolommen te verwijderen behalve enkele in Pandas