创建2个1维数组的2维数组

2024-04-26 00:00:25 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用了numpy和scipy,有些函数真正关心数组的维数,我有一个函数名CovexHull(point),它接受点作为二维数组。

hull = ConvexHull(points)

In [1]: points.ndim
Out[1]: 2
In [2]: points.shape
Out[2]: (10, 2)
In [3]: points
Out[3]: 
array([[ 0. ,  0. ],
       [ 1. ,  0.8],
       [ 0.9,  0.8],
       [ 0.9,  0.7],
       [ 0.9,  0.6],
       [ 0.8,  0.5],
       [ 0.8,  0.5],
       [ 0.7,  0.5],
       [ 0.1,  0. ],
       [ 0. ,  0. ]])

正如你在上面看到的,这些点是一个带有ndim 2的numpy。

现在我有了两个不同的numpy数组(tp和fp),如下所示(例如fp)

In [4]: fp.ndim
Out[4]: 1
In [5]: fp.shape
Out[5]: (10,)
In [6]: fp
Out[6]: 
array([ 0. ,  0.1,  0.2,  0.3,  0.4,  0.4,
        0.5, 0.6,  0.9,  1. ])

我的问题是如何创建一个二维numpy数组,就像tp和fp的点一样有效


Tags: 函数innumpyscipy数组outarraypoints
2条回答

如果你想把两个10个元素的一维数组组合成一个二维数组np.vstack((tp, fp)).T就可以了。np.vstack((tp, fp))将返回一个shape数组(2,10),而T属性返回带有shape(10,2)的转置数组(即两个一维数组形成列而不是行)。

>>> tp = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> tp.ndim
1
>>> tp.shape
(10,)

>>> fp = np.array([10, 11, 12, 13, 14, 15, 16, 17, 18, 19])
>>> fp.ndim
1
>>> fp.shape
(10,)

>>> combined = np.vstack((tp, fp)).T
>>> combined
array([[ 0, 10],
       [ 1, 11],
       [ 2, 12],
       [ 3, 13],
       [ 4, 14],
       [ 5, 15],
       [ 6, 16],
       [ 7, 17],
       [ 8, 18],
       [ 9, 19]])

>>> combined.ndim
2
>>> combined.shape
(10, 2)

你可以使用numpy的column_stack

np.column_stack((tp, fp))

相关问题 更多 >