当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python cudf.DataFrame.memory_usage用法及代码示例


用法:

DataFrame.memory_usage(index=True, deep=False)

返回每列的内存使用量(以字节为单位)。内存使用可以选择包括索引和object dtype 元素的贡献。

参数

index布尔值,默认为真

指定是否在返回的 Series 中包含 DataFrame 索引的内存使用情况。如果 index=True ,则索引的内存使用量是输出中的第一项。

deep布尔值,默认为 False

如果为 True,则通过询问 object dtypes 来深入检查数据以了解系统级内存消耗,并将其包含在返回值中。

返回

Series

一个系列,其索引是原始列名,其值是每列的内存使用量(以字节为单位)。

例子

>>> dtypes = ['int64', 'float64', 'object', 'bool']
>>> data = dict([(t, np.ones(shape=5000).astype(t))
...              for t in dtypes])
>>> df = cudf.DataFrame(data)
>>> df.head()
   int64  float64  object  bool
0      1      1.0     1.0  True
1      1      1.0     1.0  True
2      1      1.0     1.0  True
3      1      1.0     1.0  True
4      1      1.0     1.0  True
>>> df.memory_usage(index=False)
int64      40000
float64    40000
object     40000
bool        5000
dtype: int64

使用分类有效存储具有许多重复值的 object-dtype 列。

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

相关用法


注:本文由纯净天空筛选整理自rapids.ai大神的英文原创作品 cudf.DataFrame.memory_usage。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。