Wednesday, December 7, 2022

Conditional DataFrame column operations

Sử dụng when và otherwise

Chẳng hạn bạn thêm một cột 'tên_côt_gioi_tinh', các giá trị trong cột này sẽ là N nếu giá trị ở cột gioi_tinh tương ứng là 'Nam', còn lại thì sẽ là G.

df = df.withColumn('ten_cot_gioi_tinh', when(df.gioi_tinh == 'Nam', N)

                                                                .otherwise(G))

No comments:

Post a Comment

Bạn cần thêm thông tin hay có câu hỏi vui lòng comment