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


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


用法:

DataFrame.select_dtypes(include=None, exclude=None)

根据列 dtypes 返回 DataFrame 列的子集。

参数

include字符串或列表

根据 dtypes 包含哪些列

exclude字符串或列表

根据 dtypes 排除哪些列

返回

DataFrame

帧的子集,包括 include 中的 dtype,不包括 exclude 中的 dtype。

抛出

ValueError
  • 如果includeexclude都为空
  • 如果 includeexclude 有重叠的元素

例子

>>> import cudf
>>> df = cudf.DataFrame({'a': [1, 2] * 3,
...                    'b': [True, False] * 3,
...                    'c': [1.0, 2.0] * 3})
>>> df
   a      b    c
0  1   True  1.0
1  2  False  2.0
2  1   True  1.0
3  2  False  2.0
4  1   True  1.0
5  2  False  2.0
>>> df.select_dtypes(include='bool')
       b
0   True
1  False
2   True
3  False
4   True
5  False
>>> df.select_dtypes(include=['float64'])
     c
0  1.0
1  2.0
2  1.0
3  2.0
4  1.0
5  2.0
>>> df.select_dtypes(exclude=['int'])
       b    c
0   True  1.0
1  False  2.0
2   True  1.0
3  False  2.0
4   True  1.0
5  False  2.0

相关用法


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