利用numpy制作一维二维阵列
利用numpy制作一维二维阵列
提问于 2020-07-16 08:42:54
我有一个数字列表，希望添加第二列，使数组变为2D，如下例所示：
a = [1,1,1,1,1]
b = [2,2,2,2,2]应成为：
c = [[1,2],[1,2],[1,2],[1,2],[1,2]]我不知道怎么用numpy做这件事？
回答 2
Stack Overflow用户
回答已采纳
发布于 2020-07-16 08:51:29
我只需要将它们堆叠起来，然后用.T转换结果数组。
import numpy as np
a = np.array([1, 1, 1, 1, 1])
b = np.array([2, 2, 2, 2, 2])
c = np.stack((a, b)).TStack Overflow用户
发布于 2020-07-16 09:11:10
使用numpy内置函数：
import numpy as np
c = np.vstack((np.array(a),np.array(b))).T.tolist()np.vstack垂直叠加数组。.T转置数组，tolist()将其转换回列表。
另一种类似的方法是使用[:,None]添加一个维度，然后您可以水平地堆叠它们，而不需要转换：
c = np.hstack((np.array(a)[:,None],np.array(b)[:,None])).tolist())产出：
[[1, 2], [1, 2], [1, 2], [1, 2], [1, 2]]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/62930860
复制相关文章
相似问题
腾讯云开发者
