如何将Python输出的数组列表转换为n×4格式的numpy矩阵？

解决方案 #

你只需要在现有逻辑基础上做少量修改，就能得到n×4格式的numpy矩阵，以下是两种可行的实现方式： #

方式1：在原有收集逻辑后增加转换步骤

你原有循环跑完得到的是存储了多个一维numpy数组的列表，只需要在最后加一行转换代码即可：

test_labels = []
for i in range(0, number_of_generator_calls):
    test_labels.extend(np.array(test_generator[i][1]))
# 新增转换逻辑，以下二选一即可
test_labels = np.array(test_labels)
# 或者用兼容性更强的堆叠写法，适配内部元素的不同维度格式
# test_labels = np.vstack(test_labels)

转换完成后test_labels.shape会返回(n, 4)，就是你需要的n×4矩阵，此时打印前3行、后3行的输出就和你期望的格式完全一致。 #

方式2：一步生成目标矩阵，简化代码

你也可以直接用列表推导加堆叠的写法，省略中间的列表收集步骤，代码更简洁：

test_labels = np.vstack([test_generator[i][1] for i in range(number_of_generator_calls)])

补充说明 #

你之前去掉np.array后输出没有变化，是因为test_generator[i][1]本身就已经是numpy数组类型，额外套一层np.array()不会改变数据本身的格式和值。

内容的提问来源于stack exchange，提问作者Yogesh Riyat