Holen Sie sich bestimmte Reihe als Serie von Pandas Datenrahmen

Question

Wie erhalten wir eine bestimmte gefilterte Reihe als Serie?

Beispiel Datenrahmen:

>>> df = pd.DataFrame({'date': [20130101, 20130101, 20130102], 'location': ['a', 'a', 'c']}) 
>>> df 
     date location 
0 20130101  a 
1 20130101  a 
2 20130102  c 

ich brauche die Zeile auszuwählen, wo location ist cals eine Serie.

Ich habe versucht:

row = df[df["location"] == "c"].head(1) # gives a dataframe 
row = df.ix[df["location"] == "c"]  # also gives a dataframe with single row 

In beiden Fällen kann ich nicht die Zeile als Serie.

Answer 1

Über die squeeze Funktion, die eine Dimension von der Datenrahmen entfernen wird:

df[df["location"] == "c"].squeeze() 
Out[5]: 
date  20130102 
location   c 
Name: 2, dtype: object 

DataFrame.squeeze Verfahren wirkt auf die gleiche Weise des Squeeze Argument der Funktion read_csv wenn auf Wahr gesetzt: wenn die resultierende Datenrahmen ist, ein 1-len Datenrahmen, dh es hat nur eine Dimension wie eine Spalte oder eine Zeile, dann wird das Objekt auf das kleinere Dimensionsobjekt gedrückt. In Ihrem Fall erhalten Sie ein Series-Objekt aus dem DataFrame. Dieselbe Logik gilt, wenn Sie ein Panel auf einen DataFrame drücken.

Squeeze ist explizit in Ihrem Code und zeigt deutlich Ihre Absicht, das Objekt in Händen "niederzuwerfen", weil seine Dimension auf eine kleinere projiziert werden kann.

Wenn der Datenrahmen mehr als eine Spalte oder Zeile enthält, hat Squeeze keine Auswirkung.

Answer 2

Sie können direkt mit der ersten Reihe mit ganzzahligen Indizierung nehmen (iloc()-Funktion):

>>> df[df["location"] == "c"].iloc[0] 
date  20130102 
location   c 
Name: 2, dtype: object