Конвертировать z.select

Question

Я хотел бы преобразовать кодирование zeppelin, поскольку я не могу использовать zeppelin в блокноте jupyter (я уже установил пакет py4j, но он не работает)

Для этого кода я хотел бы выбрать столбцы последовательно.

total_seg_options = ["col_a","col_b","col_c","col_d","col_e"]

no_seg: я ввожу 4 и хочу выбрать [col_b, col_d, col_a, col_c]

Как мне преобразовать этот код

Вот моя кодировка zeppelin:

no_seg = int(z.input("Enter the number of segments"))
dropDown=[]
for each_Segment in total_seg_options:
    dropDown.append((each_Segment,each_Segment))
seg_cond=[]
for i in range(no_seg):
    seg_cond.append(z.select("Segment_"+str(i+1),dropDown))