pandas.DataFrame.memory_usage

DataFrame.memory_usage(index=True, deep=False) [source]

Возвращает объём памяти, занимаемый каждым столбцом в байтах.

Объём памяти может включать вклад индекса и элементов типа данных object.

По умолчанию это значение отображается в DataFrame.info. Его можно скрыть, установив pandas.options.display.memory_usage в False.

Параметры:

index : bool, по умолчанию True Указывает, следует ли включать в объём памяти DataFrame’s индекса в возвращаемом Series. Если index=True объём памяти индекса, то он будет первым элементом в выводе. deep : bool, по умолчанию False Если True, глубоко исследуются данные путём запроса object типов данных для вычисления потребления памяти на системном уровне, и это значение включается в возвращаемые значения.

Возвращаемое значение:

sizes : Series Series, индекс которого — имена исходных столбцов, а значения — объём памяти каждого столбца в байтах.

См. также

numpy.ndarray.nbytes

Общий объём байтов, потребляемых элементами ndarray.

Series.memory_usage

Объём байтов, потребляемых Series.

pandas.Categorical

Экономичный массив для строковых значений с множеством повторяющихся значений.

DataFrame.info

Краткое описание DataFrame.

Примеры

>>> dtypes = ['int64', 'float64', 'complex128', 'object', 'bool']
>>> data = dict([(t, np.ones(shape=5000).astype(t))
...              for t in dtypes])
>>> df = pd.DataFrame(data)
>>> df.head()
   int64  float64  complex128 object  bool
0      1      1.0      (1+0j)      1  True
1      1      1.0      (1+0j)      1  True
2      1      1.0      (1+0j)      1  True
3      1      1.0      (1+0j)      1  True
4      1      1.0      (1+0j)      1  True

>>> df.memory_usage()
Index            80
int64         40000
float64       40000
complex128    80000
object        40000
bool           5000
dtype: int64

>>> df.memory_usage(index=False)
int64         40000
float64       40000
complex128    80000
object        40000
bool           5000
dtype: int64

По умолчанию размер памяти столбцов типа object не учитывается:

>>> df.memory_usage(deep=True)
Index             80
int64          40000
float64        40000
complex128     80000
object        160000
bool            5000
dtype: int64

Используйте Categorical для эффективного хранения столбца типа object со многими повторяющимися значениями.

>>> df['object'].astype('category').memory_usage(deep=True)
5168