如何在pyspark数据帧中显示特定的最大行数

from pyspark.sql import SparkSession
from pyspark.sql.functions import row_number
from pyspark.sql.window import Window

spark = SparkSession.builder \
    .appName("Display Specific Number of Rows in PySpark Dataframe") \
    .getOrCreate()

df = spark.read.csv("data.csv", header=True, inferSchema=True)

max_rows = 10  # 设定要显示的最大行数
window_spec = Window.orderBy("column_name")  # 根据某一列进行排序，如果不需要排序可以不设置
df.withColumn("row_number", row_number().over(window_spec)) \
  .filter(f"row_number <= {max_rows}") \
  .drop("row_number") \
  .show(truncate=False)