本文介绍了假设我有一个2D数组，并且我想将其重塑为1D，每行一个值的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
这是我的数组
arr = np.array([[0, 1],
[3, 4],
[6, 7]])
flat_arr = np.reshape(arr, -1)
我得到以下结果：
[0 1 2 3 4 5 6 7 8]
我希望得到的结果是：
[0]
[1]
[3]
[4]
[5]...
推荐答案
有几种方法：
flat_arr[:, None]
flat_arr[:, np.newaxis]
np.expand_dims(flat_arr, axis=1)
此外，您还可以按如下方式重塑它：
arr.reshape(-1, 1)
这篇关于假设我有一个2D数组，并且我想将其重塑为1D，每行一个值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！