Comment extraire les noms de colonnes contenant une chaîne spécifiée dans des dataframes ?

Extraire les noms de colonnes contenant une chaîne spécifiée

Dans les dataframes, accéder aux colonnes par des noms spécifiques peut être crucial. Cette question concerne le scénario dans lequel il faut identifier une colonne dont le nom contient une chaîne spécifique, même s'il ne s'agit pas d'une correspondance exacte. L'exemple fourni recherche « spike » dans les noms de colonnes tels que « spike-2 », « hey Spike » et « spike-in ».

Solution :

Pour y parvenir, une approche simple consiste à parcourir les colonnes du DataFrame :

<code class="python">for col in df.columns:
    if 'spike' in col:
        # Do something with the column name</code>

Dans cette solution, chaque nom de colonne est inspecté pour vérifier s'il contient la chaîne cible. Si une correspondance est trouvée, le nom de la colonne peut être stocké dans une variable pour une utilisation ultérieure.

Une autre option consiste à utiliser la compréhension et le filtrage de liste pour créer un nouveau cadre de données avec uniquement les colonnes correspondantes :

<code class="python">spike_cols = [col for col in df.columns if 'spike' in col]
df2 = df.filter(regex='spike')</code>

La première ligne génère une liste de noms de colonnes contenant « spike », tandis que la deuxième ligne filtre le dataframe pour inclure uniquement ces colonnes.

En tirant parti de ces techniques, vous pouvez identifier et accéder efficacement aux colonnes. dont les noms contiennent une chaîne spécifique, élargissant vos capacités analytiques.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!