如何轻松地将numpy.ndarray转换为numpy.array列表？

Question

我目前正在努力将一些数据解析为培训框架。

问题是框架无法处理ndarray。我需要转换成数组列表。输入和输出数据当前存储为numpy.ndarray的两个单独列表。

1. 输入数据必须转换为numpy数组列表，其中每个数组都包含ndarray的列。

2. 输出数据必须转换为numpy数组列表，其中每个数组都包含ndarray的行？..

是否可以将其转换为此？

当我打印train_output_data [0]时，我得到了这个：

https://ufile.io/fa816

Answer 1

假设ip和op分别是输入列表和输出列表，

newInput = [ip[:,i] for i in range(ip.shape[0])]
newOutput = [x for x in op]

如果train_output_data和train_input_data是2D numpy.ndarray的列表，那么替代方案可以是

newInput = []
for ip in train_input_data:
    newInput.append([ip[:,i] for i in range(ip.shape[0])])

newOutput = []
for op in train_output_data:
    newOutput.append([x for x in op])