pyspark.sql.Column.__getitem__

列。 __getitem__ ( k : Any ) → pyspark.sql.column.Column [source]

从列表中获取位置 ordinal 的项,或者从字典中通过键获取项。

在版本 1.3.0 中新增。

在 3.4.0 版本中更改: 支持 Spark Connect。

Parameters
k

一个字面值,或者一个没有步长的切片对象。

Returns

表示通过键从字典中获取的项,或通过给定切片对象切片的子字符串。

示例

>>> df = spark.createDataFrame([('abcedfg', {"key": "value"})], ["l", "d"])
>>> df.select(df.l[slice(1, 3)], df.d['key']).show()
+------------------+------+
|substring(l, 1, 3)|d[key]|
+------------------+------+
|               abc| value|
+------------------+------+