Szukaj…


Przenoszenie danych z pand do rodzimych struktur Python i 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

Pobieranie listy python z serii:

In [3]: df['A'].tolist()
Out[3]: [1, 2, 3]

DataFrames nie mają metody tolist() . Próba powoduje błąd AttributeError:

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'

Uzyskiwanie tablicy numpy z serii:

In [5]: df['B'].values
Out[5]: array([ 1.,  2.,  3.])

Możesz także uzyskać tablicę kolumn jako pojedyncze tablice numpy z całej ramki danych:

In [6]: df.values
Out[6]: 
array([[1, 1.0, 'a', True],
       [2, 2.0, 'b', False],
       [3, 3.0, 'c', True]], dtype=object)

Pobieranie słownika z serii (używa indeksu jako kluczy):

In [7]: df['C'].to_dict()
Out[7]: {0: 'a', 1: 'b', 2: 'c'}

Możesz także odzyskać całą DataFrame jako słownik:

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}}

Metoda to_dict ma kilka różnych parametrów do dostosowania sposobu formatowania słowników. Aby uzyskać listę nagrań dla każdego wiersza:

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}]

Pełna dokumentacja dostępnych opcji tworzenia słowników znajduje się w dokumentacji.



Modified text is an extract of the original Stack Overflow Documentation
Licencjonowany na podstawie CC BY-SA 3.0
Nie związany z Stack Overflow