Pythonic式的接受单个数字或列表并将其用作列表的方式(等效于numpy.atleast_1d() )

在Python中，如果你想编写一个函数，它可以接受单个数字或列表，并将其转换为一个列表（类似于numpy.atleast_1d()的功能），你可以使用以下方法：

def atleast_1d(*args):
    result = []
    for arg in args:
        if isinstance(arg, (int, float)):
            result.append([arg])
        elif isinstance(arg, list):
            result.extend(arg)
        else:
            raise TypeError("Unsupported type")
    return result

# 使用示例
print(atleast_1d(1))        # 输出: [[1]]
print(atleast_1d([1, 2, 3])) # 输出: [1, 2, 3]
print(atleast_1d(4, [5, 6])) # 输出: [[4], 5, 6]

这个函数atleast_1d使用了可变参数*args，这意味着它可以接受任意数量的参数。然后，它遍历每个参数，检查它的类型：

• 如果参数是整数或浮点数，它将其包装在一个列表中。
• 如果参数已经是列表，它将列表中的元素添加到结果列表中。
• 如果参数是其他类型，它会抛出一个TypeError。

这种方法的优点是它简单且易于理解，不需要导入额外的库，如numpy。

应用场景

这个函数可以在需要确保输入是列表的任何场景中使用，例如：

• 数据处理函数，期望输入是一组数据。
• 图形库，需要点的坐标列表。
• 数学函数，需要向量或矩阵作为输入。

参考链接

虽然这个函数是自定义的，但如果你想了解更多关于Python类型检查和可变参数的信息，可以参考以下Python官方文档：

• Python 官方文档 - 可变参数
• Python 官方文档 - 类型检查

这个函数不依赖于任何外部库，因此没有特定的第三方库链接可以提供。如果你需要一个类似于numpy.atleast_1d()的功能，并且不介意使用numpy库，你可以直接使用numpy的这个函数。