python(pyspark) for문으로 하나의 문장에 함수를 추가할 수 있나요?

조회수 27회

원래 수동으로 조작한 함수

df2 = df1.withColumn("test1", col("data.test1"))\
            .withColumn("test2", col("data.test2"))\
            .withColumn("test3", col("data.test3"))\
            .withColumn("test4", col("data.test4"))\
            .drop("data")

희망하는 상태?

for col_name in ['test1', 'test2', 'test3', 'test4']:
    df1.withColumn(col_name, col(data.col_name)).drop("data")

이런 느낌으로... 조작하려는 컬럼명의 리스트를 반복문으로 돌려 .withColumn 함수를 추가할 수 있을까요?

1 답변

답변을 하려면 로그인이 필요합니다.

Hashcode는 개발자들을 위한 무료 QnA 사이트입니다. 계정을 생성하셔야만 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)

ᕕ( ᐛ )ᕗ
로그인이 필요합니다

Hashcode는 개발자들을 위한 무료 QnA사이트 입니다. 계정을 생성하셔야만 글을 작성하실 수 있습니다.