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


Python pyspark DataFrame.astype用法及代码示例


本文简要介绍 pyspark.pandas.DataFrame.astype 的用法。

用法:

DataFrame.astype(dtype: Union[str, numpy.dtype, pandas.core.dtypes.base.ExtensionDtype, Dict[Union[Any, Tuple[Any, …]], Union[str, numpy.dtype, pandas.core.dtypes.base.ExtensionDtype]]]) → pyspark.pandas.frame.DataFrame

将 pandas-on-Spark 对象转换为指定的 dtype dtype

参数

dtype数据类型,或列名的字典 -> 数据类型

使用 numpy.dtype 或 Python 类型将整个 pandas-on-Spark 对象转换为相同类型。或者,使用 {col: dtype, …},其中 col 是列标签,dtype 是 numpy.dtype 或 Python 类型,将 DataFrame 的一个或多个列转换为 column-specific 类型。

返回

casted与调用者相同的类型

例子

>>> df = ps.DataFrame({'a': [1, 2, 3], 'b': [1, 2, 3]}, dtype='int64')
>>> df
   a  b
0  1  1
1  2  2
2  3  3

转换为浮点类型:

>>> df.astype('float')
     a    b
0  1.0  1.0
1  2.0  2.0
2  3.0  3.0

转回 int64 类型:

>>> df.astype('int64')
   a  b
0  1  1
1  2  2
2  3  3

将 a 列转换为浮点类型:

>>> df.astype({'a': float})
     a  b
0  1.0  1
1  2.0  2
2  3.0  3

相关用法


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