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

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

用法: property DataFrame.at

访问行/列标签对的单个值。如果索引不唯一，则所有匹配对都以数组形式返回。与 loc 类似，两者都提供基于标签的查找。如果您只需要获取 DataFrame 或系列中的单个值，请使用 at。

注意

与 pandas 不同，pandas-on-Spark 只允许使用 at 获取值，但不能设置它们。

注意

警告：如果row_index 匹配很多行，将获取大量数据，可能导致您的机器内存不足。

抛出：

KeyError: 当DataFrame中不存在标签时

例子：

>>> psdf = ps.DataFrame([[0, 2, 3], [0, 4, 1], [10, 20, 30]],
...                    index=[4, 5, 5], columns=['A', 'B', 'C'])
>>> psdf
    A   B   C
4   0   2   3
5   0   4   1
5  10  20  30

获取指定行/列对的值

>>> psdf.at[4, 'B']
2

如果索引出现多次，则获取数组

>>> psdf.at[5, 'B']
array([ 4, 20])

相关用法

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