python转化为整数_python – 舍入double值并转换为整数

我在PySpark中有一个数据框,如下所示.

import pyspark.sql.functions as func

df = sqlContext.createDataFrame(

[(0.0, 0.2, 3.45631),

(0.4, 1.4, 2.82945),

(0.5, 1.9, 7.76261),

(0.6, 0.9, 2.76790),

(1.2, 1.0, 9.87984)],

["col1", "col2", "col3"])

df.show()

+----+----+-------+

|col1|col2| col3|

+----+----+-------+

| 0.0| 0.2|3.45631|

| 0.4| 1.4|2.82945|

| 0.5| 1.9|7.76261|

| 0.6| 0.9| 2.7679|

| 1.2| 1.0|9.87984|

+----+----+-------+

# round 'col3' in a new column:

df2 = df.withColumn("col4", func.round(df["col3"], 2))

df2.show()

+----+----+-------+----+

|col1|col2| col3|col4|

+----+----+-------+----+

| 0.0| 0.2|3.45631|3.46|

| 0.4| 1.4|2.82945|2.83|

| 0.5| 1.9|7.76261|7.76|

| 0.6| 0.9| 2.7679|2.77|

| 1.2| 1.0|9.87984|9.88|

+----+----+-------+----+

在上面的数据框架中,col4是双倍的.现在我想将col4转换为Integer

df2 = df.withColumn("col4", func.round(df["col3"], 2).cast('integer'))

+----+----+-------+----+

|col1|col2| col3|col4|

+----+----+-------+----+

| 0.0| 0.2|3.45631| 3|

| 0.4| 1.4|2.82945| 2|

| 0.5| 1.9|7.76261| 7|

| 0.6| 0.9| 2.7679| 2|

| 1.2| 1.0|9.87984| 9|

+----+----+-------+----+

但我想将col4值四舍五入到最近

预期结果

+----+----+-------+----+

|col1|col2| col3|col4|

+----+----+-------+----+

| 0.0| 0.2|3.45631| 3|

| 0.4| 1.4|2.82945| 3|

| 0.5| 1.9|7.76261| 8|

| 0.6| 0.9| 2.7679| 3|

| 1.2| 1.0|9.87984| 10|

+----+----+-------+----+

我怎样才能做到这一点？