我有一个数字列表,希望添加第二列,使数组变为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做这件事?
发布于 2020-07-16 00: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)).T
发布于 2020-07-16 01: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]]
https://stackoverflow.com/questions/62930860
复制