Создайте количество переменных на основе количества столбцов в фрейме данных Spark

Question

У меня есть 4 значения в столбце фрейма данных Spark -

colA
Read
Join
Sort
Write

Результатом является создание переменной 4 python на основе значений столбца, таких как -

v1=Read
v2=Join
v3=Sort
v4=Write

Если есть 3 значения в столбце. Требуется создать 3 переменные.

Спасибо

Answer 1

Вы можете собирать данные в словаре и использовать значения словаря в качестве переменных везде, где это необходимо.

vars = {i:v.colA for i,v in enumerate(df.collect())}

Теперь для доступа к данным просто используйте словарь с ключом, например

print(vars[0])
[Out] : Read