python - 如何从numpy中的D数组创建D维坐标数组

Question

在 numpy 中，我有 D 个长度为 n 的数组，我想从这些数组生成一个长度为 n 的 D 维坐标数组。然后，每个数组提供坐标轴的值。

例如，在二维中:

import numpy as np
x = np.arange(5)
y = x + 4

#from which I would like to make this nd.array:
[[0, 4],
 [1, 5],
 [2, 6],
 [3, 7],
 [4, 8]]

Answer 1

您可以使用column_stack来做到这一点:

>>> np.column_stack((x, y))
array([[0, 4],
       [1, 5],
       [2, 6],
       [3, 7],
       [4, 8]])

column_stack 采用数组序列，因此如果您想要堆叠 D 数组，则可以直接使用 D 调用 column_stack。