TypeError：“DataFrame”对象在使用除法时不支持项目分配

Posted 2023-04-15

【中文标题】TypeError：“DataFrame”对象在使用除法时不支持项目分配

【英文标题】：TypeError: 'DataFrame' object does not support item assignment when using division

【发布时间】：2021-11-23 19:30:39

【问题描述】：

TypeError: 'DataFrame' 对象在使用除法时不支持项目分配。

在 pyspark 数据帧中划分两个整数列并收到此错误。有什么想法吗？

可重现：

simpleData = (((1, 2), (2, 6), (3, 6))
columns= ["sales","users"]
df = spark.createDataFrame(data = simpleData, schema = columns)
df = df.withColumn("users", df["users"].cast(IntegerType()))
df = df.withColumn("sales", df["sales"].cast(IntegerType()))
df['buy_rate'] = df['sales']/ df['users']

【参考方案1】：

只是小调整。

from pyspark.sql.types import IntegerType
import pyspark.sql.functions as F
simpleData=((1, 2), (2, 6), (3, 6))
columns= ["sales","users"]
df = spark.createDataFrame(data = simpleData, schema = columns)
df = df.withColumn("users", df["users"].cast(IntegerType()))
df = df.withColumn("sales", df["sales"].cast(IntegerType()))

df1 = df\
    .withColumn("buy_rate", (F.col("sales") / F.col("users")))
display(df1)

【讨论】：

感谢 Karthik，在 withColumn() 中分配给新列的方法非常好，效果很好。

最好的祝福格雷格！！