Python分段函数处理列表时仅执行首个元素问题求助

我编写了一个Python分段函数，测试单个输入项时运行完全正常，但传入列表作为输入时，仅会处理列表中的第一个元素。以下是我的代码：

def pwfun(x):
    myList = []
    
    if isinstance(x, list):
        for i in x:
            if i < -1:
                if isinstance(i, float):
                    myList.append(round(-2 * (i + 1), 2))
                else:
                    myList.append(int(-2 * (i + 1)))
            elif i <= 1 and i >= -1:
                myList.append(0)
            else:
                if isinstance(i, float):
                    myList.append(round(i**2 - 1, 2))
                else: 
                    mylist.append(int(i**2 - 1))
            return myList
    else:
        if x < -1:
            if isinstance(x, float):
                return(round(-2 * (x + 1), 2))
            else: 
                return(int(-2 * (x + 1)))
        elif x <= 1 and x >= -1:
            return(0)
        else:
            if isinstance(x, float):
                return(round(x**2 - 1, 2))
            else:
                return(int(x**2 -1))
   
x = [-6.2, 0]
pwfun(x)

问题分析 #

• 缩进错误：return myList写在了for循环内部，导致循环处理完第一个元素后就直接返回列表，后续元素根本没机会被执行。必须把这行代码移到for循环外面，等所有元素处理完成后再返回结果。
• 变量名拼写错误：else分支里的mylist.append是小写的m，但前面定义的列表是myList（Python对大小写敏感），这会触发未定义变量的报错，需要统一改成myList.append。

修正后的代码 #

def pwfun(x):
    myList = []
    
    if isinstance(x, list):
        for i in x:
            if i < -1:
                if isinstance(i, float):
                    myList.append(round(-2 * (i + 1), 2))
                else:
                    myList.append(int(-2 * (i + 1)))
            elif -1 <= i <= 1:
                myList.append(0)
            else:
                if isinstance(i, float):
                    myList.append(round(i**2 - 1, 2))
                else: 
                    myList.append(int(i**2 - 1))
        # 移到循环外，确保所有元素处理完成后返回
        return myList
    else:
        if x < -1:
            return round(-2 * (x + 1), 2) if isinstance(x, float) else int(-2 * (x + 1))
        elif -1 <= x <= 1:
            return 0
        else:
            return round(x**2 - 1, 2) if isinstance(x, float) else int(x**2 - 1)
   
x = [-6.2, 0]
print(pwfun(x))  # 输出: [10.4, 0]

代码优化（可选） #

把单个元素的处理逻辑抽成子函数，避免重复代码，让结构更清晰：

def process_single(x):
    if x < -1:
        return round(-2 * (x + 1), 2) if isinstance(x, float) else int(-2 * (x + 1))
    elif -1 <= x <= 1:
        return 0
    else:
        return round(x**2 - 1, 2) if isinstance(x, float) else int(x**2 - 1)

def pwfun(x):
    if isinstance(x, list):
        return [process_single(i) for i in x]
    else:
        return process_single(x)

x = [-6.2, 0, 3]
print(pwfun(x))  # 输出: [10.4, 0, 8]

内容的提问来源于stack exchange，提问作者Joseph Silva