是否存在C API函数PyArray_ContiguousFromAny的Python等价实现？

你可以通过先检查维度再转换为连续数组的方式实现和PyArray_ContiguousFromAny等价的功能，避免不必要的内存复制。具体思路是先用np.asarray()（不强制复制）获取数组对象，验证维度范围后，再用np.ascontiguousarray()处理连续性，这样只有维度符合要求时才会触发可能的内存复制。

封装实现示例 #

import numpy as np

def contiguous_from_any(op, dtype=None, min_depth=0, max_depth=np.inf):
    # 先将输入转为numpy数组（不强制复制）
    arr = np.asarray(op, dtype=dtype)
    # 检查维度是否在指定范围内
    if not (min_depth <= arr.ndim <= max_depth):
        raise ValueError(f"数组维度{arr.ndim}不在[{min_depth}, {max_depth}]范围内")
    # 转换为连续数组（仅当原数组不连续时才复制）
    return np.ascontiguousarray(arr, dtype=dtype)

关键细节说明 #

• np.asarray(op, dtype=dtype)：仅在输入不是numpy数组时才会创建新数组，否则直接返回原数组的视图，不会产生额外内存开销，这一步先完成类型转换和数组包装。
• 维度检查：在进行连续化操作前先验证arr.ndim，不符合要求直接抛出异常，避免后续不必要的内存复制。
• np.ascontiguousarray(arr, dtype=dtype)：只有当原数组的内存布局不是C连续时，才会创建新的连续数组；如果已经是连续的，直接返回原数组的引用，性能开销极小。

这个实现完全对应PyArray_ContiguousFromAny的核心逻辑：类型转换、维度校验、连续化处理，同时避免了先连续化再检查维度的资源浪费。

内容的提问来源于stack exchange，提问作者Frank Yellin