首页 > 解决方案 > 我怎样才能像坐标一样排列这个输出

问题描述

我有这样的输出

array([[-10], [-23], [-12]])

但我需要它是这样的

 array([-10,  -23, -12])

已经尝试过 np.reshape()。

谢谢你

标签: pythonnew-operator

解决方案


使用flatten

import numpy as np
a = np.array([[-10], [-23], [-12]])
a.flatten()

输出:

array([-10, -23, -12])

注意:如果您只有以下列表,请根据您的评论:

lst = [[-10], [-23], [-12]]

您需要首先将其转换为 numpy 数组(根据您的问题),您可以通过以下方式执行此操作:

a = np.array(lst)

然后通过以下方式将其展平:

a.flatten()

推荐阅读