pandas
Faire jouer les Pandas avec les types de données Python natifs
Recherche…
Déplacement de données hors de pandas vers des structures de données natives Python et Numpy
In [1]: df = pd.DataFrame({'A': [1, 2, 3], 'B': [1.0, 2.0, 3.0], 'C': ['a', 'b', 'c'],
'D': [True, False, True]})
In [2]: df
Out[2]:
A B C D
0 1 1.0 a True
1 2 2.0 b False
2 3 3.0 c True
Obtenir une liste de python à partir d'une série:
In [3]: df['A'].tolist()
Out[3]: [1, 2, 3]
Les DataFrames n'ont pas de méthode tolist()
. Son essai entraîne une erreur d'attribut:
In [4]: df.tolist()
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-4-fc6763af1ff7> in <module>()
----> 1 df.tolist()
//anaconda/lib/python2.7/site-packages/pandas/core/generic.pyc in __getattr__(self, name)
2742 if name in self._info_axis:
2743 return self[name]
-> 2744 return object.__getattribute__(self, name)
2745
2746 def __setattr__(self, name, value):
AttributeError: 'DataFrame' object has no attribute 'tolist'
Obtenir un tableau numpy d'une série:
In [5]: df['B'].values
Out[5]: array([ 1., 2., 3.])
Vous pouvez également obtenir un tableau des colonnes sous forme de tableaux numpy individuels à partir d'un cadre de données complet:
In [6]: df.values
Out[6]:
array([[1, 1.0, 'a', True],
[2, 2.0, 'b', False],
[3, 3.0, 'c', True]], dtype=object)
Obtenir un dictionnaire à partir d'une série (utilise l'index comme clés):
In [7]: df['C'].to_dict()
Out[7]: {0: 'a', 1: 'b', 2: 'c'}
Vous pouvez également récupérer l'intégralité de DataFrame en tant que dictionnaire:
In [8]: df.to_dict()
Out[8]:
{'A': {0: 1, 1: 2, 2: 3},
'B': {0: 1.0, 1: 2.0, 2: 3.0},
'C': {0: 'a', 1: 'b', 2: 'c'},
'D': {0: True, 1: False, 2: True}}
La méthode to_dict
a quelques paramètres différents pour ajuster le format des dictionnaires. Pour obtenir une liste de dicts pour chaque ligne:
In [9]: df.to_dict('records')
Out[9]:
[{'A': 1, 'B': 1.0, 'C': 'a', 'D': True},
{'A': 2, 'B': 2.0, 'C': 'b', 'D': False},
{'A': 3, 'B': 3.0, 'C': 'c', 'D': True}]
Voir la documentation pour la liste complète des options disponibles pour créer des dictionnaires.
Modified text is an extract of the original Stack Overflow Documentation
Sous licence CC BY-SA 3.0
Non affilié à Stack Overflow