Tương tác với DataFrame PySpark

Nền tảng Big Data với PySpark

Upendra Devisetty

Science Analyst, CyVerse

Toán tử DataFrame trong PySpark

Thao tác DataFrame: Transformations và Actions
Transformations của DataFrame:
- select(), filter(), groupby(), orderby(), dropDuplicates(), và withColumnRenamed()
Actions của DataFrame:
- printSchema(), head(), show(), count(), columns, và describe()
Đính chính: printSchema() là phương thức cho mọi Spark dataset/dataframe, không phải action

Thao tác select() và show()

Biến đổi select() chọn tập con các cột trong DataFrame

df_id_age = test.select('Age')

Hành động show() in 20 hàng đầu của DataFrame

df_id_age.show(3)

+---+
|Age|
+---+
| 17|
| 17|
| 17|
+---+
only showing top 3 rows

Thao tác filter() và show()

Biến đổi filter() lọc các hàng theo điều kiện

new_df_age21 = new_df.filter(new_df.Age > 21)
new_df_age21.show(3)

+-------+------+---+
|User_ID|Gender|Age|
+-------+------+---+
|1000002|     M| 55|
|1000003|     M| 26|
|1000004|     M| 46|
+-------+------+---+
only showing top 3 rows

Thao tác groupby() và count()

Thao tác groupby() dùng để nhóm theo một biến

test_df_age_group = test_df.groupby('Age')
test_df_age_group.count().show(3)

+---+------+
|Age| count|
+---+------+
| 26|219587|
| 17|     4|
| 55| 21504|
+---+------+
only showing top 3 rows

Biến đổi orderby()

Thao tác orderby() sắp xếp DataFrame theo một hoặc nhiều cột

test_df_age_group.count().orderBy('Age').show(3)

+---+-----+
|Age|count|
+---+-----+
|  0|15098|
| 17|    4|
| 18|99660|
+---+-----+
only showing top 3 rows

dropDuplicates()

dropDuplicates() loại bỏ các hàng trùng lặp của DataFrame

test_df_no_dup = test_df.select('User_ID','Gender', 'Age').dropDuplicates()
test_df_no_dup.count()

Biến đổi withColumnRenamed

withColumnRenamed() đổi tên một cột trong DataFrame

test_df_sex = test_df.withColumnRenamed('Gender', 'Sex')
test_df_sex.show(3)

+-------+---+---+
|User_ID|Sex|Age|
+-------+---+---+
|1000001|  F| 17|
|1000001|  F| 17|
|1000001|  F| 17|
+-------+---+---+

printSchema()

Thao tác printSchema() in kiểu dữ liệu của các cột trong DataFrame

test_df.printSchema()

 |-- User_ID: integer (nullable = true)
 |-- Product_ID: string (nullable = true)
 |-- Gender: string (nullable = true)
 |-- Age: string (nullable = true)
 |-- Occupation: integer (nullable = true)
 |-- Purchase: integer (nullable = true)

Hành động columns

Toán tử columns in tên các cột của DataFrame

test_df.columns

['User_ID', 'Gender', 'Age']

Hành động describe()

Thao tác describe() tính thống kê tóm tắt cho các cột số trong DataFrame

test_df.describe().show()

+-------+------------------+------+------------------+
|summary|           User_ID|Gender|               Age|
+-------+------------------+------+------------------+
|  count|            550068|550068|            550068|
|   mean|1003028.8424013031|  null|30.382052764385495|
| stddev|1727.5915855307312|  null|11.866105189533554|
|    min|           1000001|     F|                 0|
|    max|           1006040|     M|                55|
+-------+------------------+------+------------------+

Ayo berlatih!

Nền tảng Big Data với PySpark