计算Pyspark数据帧中的运行总数，并在出现条件时中断循环

from pyspark.sql import SparkSession
from pyspark.sql.functions import col

spark = SparkSession.builder.appName("DataFrame Count").getOrCreate()

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

count = df.count()

if count > 1000:
    raise Exception("Count exceeds 1000. Stopping the loop.")

from pyspark.sql import SparkSession
from pyspark.sql.functions import col

spark = SparkSession.builder.appName("DataFrame Count").getOrCreate()

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

count = df.count()

if count > 1000:
    raise Exception("Count exceeds 1000. Stopping the loop.")