如何在pyspark中只保存价值

Question

我将 pyspark 数据帧中的列的值保存到数据帧仅包含一列的变量中：

variable=df.select(df['columnA']).collect()
print(variable)

输出：

[Row(columnA='value')]

但我希望变量只包含“值”如何实现？

Answer 1

试试下面的代码：

import pandas as pd

df = pd.read_csv(file_name)

variable_name = df[column_name]

Answer 2

# Retrieving data from the "columnA" column
  for col in df.collect():
    print(col["columnA"])


# first row - first column
print(df.collect()[0][0])