如何将单个numpy数组内的两个元素拆分为独立数组

实现方法 #

你可以根据使用场景选择以下两种方案： #

方案1：直接索引拆分（适合固定2个元素的场景）

直接提取原数组的两个元素，分别封装为新的numpy数组即可：

import numpy as np

# 原始输入
input_data = [np.array([0.1, 5.6])]

# 拆分逻辑
result = [
    np.array([input_data[0][0]]),
    np.array([input_data[0][1]])
]

print(result)
# 输出：[array([0.1]), array([5.6])]

方案2：通用拆分方案（适合任意长度的原数组）

用numpy内置的np.split方法，自动按元素数量拆分，不需要手动写索引：

import numpy as np

input_data = [np.array([0.1, 5.6])]

# 按元素个数拆分，直接转成列表即可
result = list(np.split(input_data[0], len(input_data[0])))

print(result)
# 输出：[array([0.1]), array([5.6])]

内容的提问来源于stack exchange，提问作者glenn golez