Pyspark創建時間戳列

Question

我正在使用spark 2.1.0。我無法在pyspark中創建時間戳列，我正在使用下面的代碼段。請幫助

df=df.withColumn('Age',lit(datetime.now())) 

我越來越

assertion error:col should be Column

請幫

Answer 1

假設您從您的代碼段有數據框中，你想對所有的行相同的時間戳。

讓我創建一些虛擬數據框。

>>> dict = [{'name': 'Alice', 'age': 1},{'name': 'Again', 'age': 2}] 
>>> df = spark.createDataFrame(dict) 

>>> import time 
>>> import datetime 
>>> timestamp = datetime.datetime.fromtimestamp(time.time()).strftime('%Y-%m-%d %H:%M:%S') 
>>> type(timestamp) 
<class 'str'> 

>>> from pyspark.sql.functions import lit,unix_timestamp 
>>> timestamp 
'2017-08-02 16:16:14' 
>>> new_df = df.withColumn('time',unix_timestamp(lit(timestamp),'yyyy-MM-dd HH:mm:ss').cast("timestamp")) 
>>> new_df.show(truncate = False) 
+---+-----+---------------------+ 
|age|name |time     | 
+---+-----+---------------------+ 
|1 |Alice|2017-08-02 16:16:14.0| 
|2 |Again|2017-08-02 16:16:14.0| 
+---+-----+---------------------+ 

>>> new_df.printSchema() 
root 
|-- age: long (nullable = true) 
|-- name: string (nullable = true) 
|-- time: timestamp (nullable = true)