Skip to content

Pandas How to Find-Spalte enthält einen bestimmten Wert

In diesem Tutorial lernen wir, wie man den Index einer Spalte findet, die einen bestimmten Wert enthält.

Beispiel: Wir haben einen Datenrahmen mit den folgenden Spalten:

import pandas as pd
df = pd.DataFrame({
    'A': [1, 4, 7, 1, 4],
    'B': [2, 5, 8, 2, 5],
    'C': [3, 6, 9, 3, 6]
})
    A   B   C
0   1   2   3
1   4   5   6
2   7   8   9
4   1   2   3
5   4   5   6

Lassen Sie uns den Index der Spalte finden, die den Wert enthält 5.

Verwenden von np.where von numpy

Die numpy-Funktion np.where kann verwendet werden, um den Index einer Spalte zu finden, die einen bestimmten Wert enthält. Die Syntax lautet:
numpy.where(condition[, x, y])

Parameters
conditionarray_like, bool
Where True, yield x, otherwise yield y.

x, y: array_like
Values from which to choose. x, y and condition need to be broadcastable to some shape.

Returns
out: ndarray
An array with elements from x where condition is True, and elements from y elsewhere.

Für das obige Beispiel können wir den folgenden Code verwenden, um den Index der Spalte zu finden B das den Wert enthält 5.

import numpy as np
col_index = pd.DataFrame(np.where(df.eq(5))[1] + 1, columns=['col_index'])

Ausgabe:

   col_index
0          2
1          2
Click to rate this post!
[Total: 0 Average: 0]



Anderer Beitrag

Schreibe einen Kommentar

Deine E-Mail-Adresse wird nicht veröffentlicht.