Как удалить столбец из строки в Dataframe на основе условия?

Question

Я хочу удалить столбец из строк, когда значение столбца в этой строке равно нулю. Я не хочу удалять столбец из Dataframe. Я хочу удалить столбец из этой конкретной строки (строк), только если значение столбца равно нулю. Я использую Pyspark.

Answer 1

Вы можете отфильтровать записи, имеющие значение 0, следующим образом

import org.apache.sql.functions.{col, when}
df.filter(col(colName).=!=0)

, где df - ваш DataFrame