Konwertuj ciąg pyspark na format daty


80

Mam ramkę danych pyspark daty z kolumną typu string w formacie MM-dd-yyyyi próbuję przekonwertować ją na kolumnę daty.

Próbowałem:

df.select(to_date(df.STRING_COLUMN).alias('new_date')).show()

i otrzymuję ciąg wartości null. Czy ktoś może pomóc?


O ile nie używasz jednego z dodatków TimeSeriesRDD (zobacz konferencję Spark 2016 w celu omówienia, są dwa, o których wiem, ale oba są nadal w fazie rozwoju), nie ma wielu świetnych narzędzi do szeregów czasowych. W związku z tym stwierdziłem, że rzadko istnieje powód, aby zawracać sobie głowę konwertowaniem ciągów znaków na obiekty typu data-godzina, jeśli Twoim celem są prawdziwe typy groupBylub operacje ponownego próbkowania. Po prostu wykonaj je na kolumnach strun.
Jeff

Analiza zostanie przeprowadzona przy użyciu niewielkich lub żadnych, groupByale raczej podłużnych badań dokumentacji medycznej. Dlatego ważna jest możliwość manipulowania datą
Jenks

Odpowiedzi:


114

Aktualizacja (10.01.2018):

W przypadku platformy Spark 2.2+ najlepszym sposobem na to jest prawdopodobnie użycie funkcji to_datelub to_timestamp, które obsługują formatargument. Z dokumentów:

>>> from pyspark.sql.functions import to_timestamp
>>> df = spark.createDataFrame([('1997-02-28 10:30:00',)], ['t'])
>>> df.select(to_timestamp(df.t, 'yyyy-MM-dd HH:mm:ss').alias('dt')).collect()
[Row(dt=datetime.datetime(1997, 2, 28, 10, 30))]

Oryginalna odpowiedź (dla Spark <2.2)

Jest możliwe (preferowane?) Zrobienie tego bez udf:

from pyspark.sql.functions import unix_timestamp, from_unixtime

df = spark.createDataFrame(
    [("11/25/1991",), ("11/24/1991",), ("11/30/1991",)], 
    ['date_str']
)

df2 = df.select(
    'date_str', 
    from_unixtime(unix_timestamp('date_str', 'MM/dd/yyy')).alias('date')
)

print(df2)
#DataFrame[date_str: string, date: timestamp]

df2.show(truncate=False)
#+----------+-------------------+
#|date_str  |date               |
#+----------+-------------------+
#|11/25/1991|1991-11-25 00:00:00|
#|11/24/1991|1991-11-24 00:00:00|
#|11/30/1991|1991-11-30 00:00:00|
#+----------+-------------------+

3
To jest poprawna odpowiedź. Użycie do tego celu udf zniszczy Twoją wydajność.
gberger

8
from pyspark.sql.functions import from_unixtime, unix_timestamp
Quetzalcoatl

Zauważ, że możesz znaleźć odniesienie do formatu daty w Javie tutaj: docs.oracle.com/javase/6/docs/api/java/text/…
RobinL

3
Zauważ również, że to_date()argumentem format jest spark 2.2+. to_dateistniał przed 2.2, ale opcja formatu nie istniała
RobinL

41
from datetime import datetime
from pyspark.sql.functions import col, udf
from pyspark.sql.types import DateType



# Creation of a dummy dataframe:
df1 = sqlContext.createDataFrame([("11/25/1991","11/24/1991","11/30/1991"), 
                            ("11/25/1391","11/24/1992","11/30/1992")], schema=['first', 'second', 'third'])

# Setting an user define function:
# This function converts the string cell into a date:
func =  udf (lambda x: datetime.strptime(x, '%m/%d/%Y'), DateType())

df = df1.withColumn('test', func(col('first')))

df.show()

df.printSchema()

Oto wynik:

+----------+----------+----------+----------+
|     first|    second|     third|      test|
+----------+----------+----------+----------+
|11/25/1991|11/24/1991|11/30/1991|1991-01-25|
|11/25/1391|11/24/1992|11/30/1992|1391-01-17|
+----------+----------+----------+----------+

root
 |-- first: string (nullable = true)
 |-- second: string (nullable = true)
 |-- third: string (nullable = true)
 |-- test: date (nullable = true)

6
A udfnie powinno być tutaj konieczne, ale wbudowane funkcje do obsługi tego są okropne. Na razie to też bym zrobił.
Jeff

3
Dlaczego daty w kolumnie testowej nie zgadzają się z pierwszą kolumną? Tak, teraz jest to typ daty, ale dni i miesiące nie są zgodne. Czy jest powód?
Jenks

1
test podaje nieprawidłowe wartości daty. To nie jest dobra odpowiedź.
Shehryar

1
Każde rozwiązanie z UDF nie jest odpowiedzią, ledwie obejściem. Nie sądzę, jest wiele przypadków użycia, których nie można zrobić, łącząc sam PSF i .transform ().
sumon c

28

Podejście strptime () nie działa dla mnie. Otrzymuję inne czystsze rozwiązanie, używając odlewu:

from pyspark.sql.types import DateType
spark_df1 = spark_df.withColumn("record_date",spark_df['order_submitted_date'].cast(DateType()))
#below is the result
spark_df1.select('order_submitted_date','record_date').show(10,False)

+---------------------+-----------+
|order_submitted_date |record_date|
+---------------------+-----------+
|2015-08-19 12:54:16.0|2015-08-19 |
|2016-04-14 13:55:50.0|2016-04-14 |
|2013-10-11 18:23:36.0|2013-10-11 |
|2015-08-19 20:18:55.0|2015-08-19 |
|2015-08-20 12:07:40.0|2015-08-20 |
|2013-10-11 21:24:12.0|2013-10-11 |
|2013-10-11 23:29:28.0|2013-10-11 |
|2015-08-20 16:59:35.0|2015-08-20 |
|2015-08-20 17:32:03.0|2015-08-20 |
|2016-04-13 16:56:21.0|2016-04-13 |

7
Dzięki, to podejście zadziałało dla mnie! Na wypadek, gdyby ktoś chciał przekonwertować ciąg, taki jak 2008-08-01T14:45:37Zznacznik czasu zamiast daty, df = df.withColumn("CreationDate",df['CreationDate'].cast(TimestampType()))działa dobrze ... (Spark 2.2.0)
Gaurav

1
Wypróbowałem tę opcję spośród wielu z AWS Glue pyspark, działa jak urok!
Abhi

11

W aktualizacji zaakceptowanej odpowiedzi nie widzisz przykładu dla to_datefunkcji, więc innym rozwiązaniem wykorzystującym ją byłoby:

from pyspark.sql import functions as F

df = df.withColumn(
            'new_date',
                F.to_date(
                    F.unix_timestamp('STRINGCOLUMN', 'MM-dd-yyyy').cast('timestamp')))

1
wykonanie prostej metody to_date () nie działa, to jest poprawna odpowiedź
ski_squaw

6

prawdopodobnie nie tak wiele odpowiedzi, więc myślę o udostępnieniu mojego kodu, który może komuś pomóc

from pyspark.sql import SparkSession
from pyspark.sql.functions import to_date

spark = SparkSession.builder.appName("Python Spark SQL basic example")\
    .config("spark.some.config.option", "some-value").getOrCreate()


df = spark.createDataFrame([('2019-06-22',)], ['t'])
df1 = df.select(to_date(df.t, 'yyyy-MM-dd').alias('dt'))
print df1
print df1.show()

wynik

DataFrame[dt: date]
+----------+
|        dt|
+----------+
|2019-06-22|
+----------+

powyższy kod, aby przekonwertować na datę, jeśli chcesz przekonwertować datę i godzinę, a następnie użyj to_timestamp. daj mi znać, jeśli masz jakiekolwiek wątpliwości.


1

Spróbuj tego:

df = spark.createDataFrame([('2018-07-27 10:30:00',)], ['Date_col'])
df.select(from_unixtime(unix_timestamp(df.Date_col, 'yyyy-MM-dd HH:mm:ss')).alias('dt_col'))
df.show()
+-------------------+  
|           Date_col|  
+-------------------+  
|2018-07-27 10:30:00|  
+-------------------+  

7
Możesz rozważyć wyjaśnienie, w jaki sposób Twoja odpowiedź poprawia to, co zostało już przekazane i zaakceptowane.
chb
Korzystając z naszej strony potwierdzasz, że przeczytałeś(-aś) i rozumiesz nasze zasady używania plików cookie i zasady ochrony prywatności.
Licensed under cc by-sa 3.0 with attribution required.