Interagire con i DataFrame con PySpark SQL

Fondamenti di Big Data con PySpark

Upendra Devisetty

Science Analyst, CyVerse

DataFrame API vs query SQL

  • In PySpark puoi usare SparkSQL tramite DataFrame API e query SQL

  • La DataFrame API offre un linguaggio DSL specifico per manipolare i dati

  • Le trasformazioni e le azioni sui DataFrame sono più semplici da scrivere in codice

  • Le query SQL possono essere concise, facili da capire e portabili

  • Le stesse operazioni sui DataFrame si possono fare anche con query SQL

Fondamenti di Big Data con PySpark

Eseguire query SQL

  • Il metodo sql() di SparkSession esegue una query SQL

  • Il metodo sql() prende un'istruzione SQL e restituisce il risultato come DataFrame

df.createOrReplaceTempView("table1")
df2 = spark.sql("SELECT field1, field2 FROM table1")
df2.collect()
[Row(f1=1, f2='row1'), Row(f1=2, f2='row2'), Row(f1=3, f2='row3')]
Fondamenti di Big Data con PySpark

Query SQL per estrarre dati

test_df.createOrReplaceTempView("test_table")
query = '''SELECT Product_ID FROM test_table'''
test_product_df = spark.sql(query)
test_product_df.show(5)
+----------+
|Product_ID|
+----------+
| P00069042|
| P00248942|
| P00087842|
| P00085442|
| P00285442|
+----------+
Fondamenti di Big Data con PySpark

Sintesi e raggruppamento con query SQL

test_df.createOrReplaceTempView("test_table")
query = '''SELECT Age, max(Purchase) FROM test_table GROUP BY Age'''
spark.sql(query).show(5)
+-----+-------------+
|  Age|max(Purchase)|
+-----+-------------+
|18-25|        23958|
|26-35|        23961|
| 0-17|        23955|
|46-50|        23960|
|51-55|        23960|
+-----+-------------+
only showing top 5 rows
Fondamenti di Big Data con PySpark

Filtrare colonne con query SQL

test_df.createOrReplaceTempView("test_table")
query = '''SELECT Age, Purchase, Gender FROM test_table WHERE Purchase > 20000 AND Gender == "F"'''
spark.sql(query).show(5)
+-----+--------+------+
|  Age|Purchase|Gender|
+-----+--------+------+
|36-45|   23792|     F|
|26-35|   21002|     F|
|26-35|   23595|     F|
|26-35|   23341|     F|
|46-50|   20771|     F|
+-----+--------+------+
only showing top 5 rows
Fondamenti di Big Data con PySpark

Esercitiamoci!

Fondamenti di Big Data con PySpark

Preparing Video For Download...