여러 열을 추가하거나 이름이 같은 기존 열을 바꿔 새 DataFrame을 반환합니다.
문법
withColumns(*colsMap: Dict[str, Column])
매개 변수
| 매개 변수 | 유형 | 설명 |
|---|---|---|
colsMap |
사전 | 열 이름 및 열의 받아쓰기입니다. 현재 단일 맵만 지원됩니다. |
Returns
DataFrame: 새 열 또는 교체된 열이 있는 데이터 프레임입니다.
예제
df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
df.withColumns({'age2': df.age + 2, 'age3': df.age + 3}).show()
# +---+-----+----+----+
# |age| name|age2|age3|
# +---+-----+----+----+
# | 2|Alice| 4| 5|
# | 5| Bob| 7| 8|
# +---+-----+----+----+