python - 除了加载数据之外如何写尝试

Question

我对编码还很陌生，所以对于这个愚蠢的问题我深表歉意。我正在编写一个 spark 函数，它接受文件路径和文件类型并创建一个数据框。如果输入无效，我只想打印某种错误消息并返回一个空数据框。我会使用 try except 吗？

def rdf(name, type):
   try:
      df=spark.read.format(type).load(name)
      return df
   except ____ as error:
      print(error)
      return "" #I want to return an empty RDD here, but I can't figure out how to make one

我怎么知道 ____ 里有什么？我尝试了 org.apache.spark.SparkException，因为这是我在将 .csv 文件作为 Parquet 传递时遇到的错误，它中断了，但它不起作用

Answer 1

欢迎来到 StackOverflow!您可以在 try-except block 中捕获多个异常；例如:

def rdf(name, type):
   try:
      df=spark.read.format(type).load(name)
      return df
   except (SparkException, TypeError) as error:
      print(error)
      return ""

您可以在该元组中替换或添加错误。

使用 Exception 可能会消除与您的代码无关的错误(例如，如果名称是 S3 路径，则会出现网络问题)。这可能是您希望您的程序不处理的事情。