我们可以使用Python中的try-except语句来检查输入是否为JSON或YAML。具体实现可参考以下代码：

import json import yaml

def is_json_or_yaml(input_data): """ Check if the input is JSON or YAML format """ try: json.loads(input_data) return "input is JSON" except ValueError: try: yaml.safe_load(input_data) return "input is YAML" except yaml.YAMLError: return "invalid input"

Example Usage

input_data = '{"name": "John", "age": 30}' print(is_json_or_yaml(input_data)) # output: input is JSON

input_data = 'name: John\nage: 30' print(is_json_or_yaml(input_data)) # output: input is YAML

input_data = 'invalid input' print(is_json_or_yaml(input_data)) # output: invalid input

该方法首先尝试以JSON格式解析输入数据，如果成功则返回"input is JSON"；如果无法解析，就尝试以YAML格式解析数据，如果成功则返回"input is YAML"；如果两种格式都无法解析，则返回"invalid input"。这种方法可以检查输入格式并提供有用的反馈。