web-dev-qa-db-fra.com

Comment obtenir la première colonne d'un DataFrame de pandas en tant que série?

J'ai essayé:

x=pandas.DataFrame(...)
s = x.take([0], axis=1)

Et s obtient un DataFrame, pas une série.

pythondataframepandasseries
98
Yariv
>>> import pandas as pd
>>> df = pd.DataFrame({'x' : [1, 2, 3, 4], 'y' : [4, 5, 6, 7]})
>>> df
   x  y
0  1  4
1  2  5
2  3  6
3  4  7
>>> s = df.ix[:,0]
>>> type(s)
<class 'pandas.core.series.Series'>
>>>

============================================== =========================

METTRE &AGRAVE; JOUR

Si vous lisez ceci après juin 2017, ix est déconseillé dans les pandas 0.20.2, alors ne l'utilisez pas. Utilisez plutôt loc ou iloc. Voir les commentaires et autres réponses à cette question.

103
herrfz

Vous pouvez obtenir la première colonne en tant que série en utilisant le code suivant:

x[x.columns[0]]
86
HYRY

À partir de v0.11 +, ... utilisez df.iloc.

In [7]: df.iloc[:,0]
Out[7]: 
0    1
1    2
2    3
3    4
Name: x, dtype: int64
65
Jeff

N'est-ce pas la façon la plus simple?

Par nom de colonne:

In [20]: df = pd.DataFrame({'x' : [1, 2, 3, 4], 'y' : [4, 5, 6, 7]})
In [21]: df
Out[21]:
    x   y
0   1   4
1   2   5
2   3   6
3   4   7

In [23]: df.x
Out[23]:
0    1
1    2
2    3
3    4
Name: x, dtype: int64

In [24]: type(df.x)
Out[24]:
pandas.core.series.Series
11
SamJ

Cela fonctionne très bien lorsque vous souhaitez charger une série à partir d'un fichier csv.

x = pd.read_csv('x.csv', index_col=False, names=['x'],header=None).iloc[:,0]
print(type(x))
print(x.head(10))


<class 'pandas.core.series.Series'>
0    110.96
1    119.40
2    135.89
3    152.32
4    192.91
5    177.20
6    181.16
7    177.30
8    200.13
9    235.41
Name: x, dtype: float64
0
Christopher Pfeifer