當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。