Python：将并行数组重塑为训练集

Question

我有两个并行数组，使用 meshgrid 创建：

X, Y = np.meshgrid(X, Y)

现在我想将其转换为NN的（Xn * Yn，2）形训练集：

[[x_1, y_1], [x_2, y_2], ..., [x_m, y_m]]

(where m = Xn * Yn)

我该怎么做？

Answer 1

您可以尝试使用reshape和stack，使用reshape函数将X和Y转换为一列2D数组（X（1，X），X），（Yn， 1）对于Y）首先然后将它们水平堆叠：

X, Y = np.meshgrid([1,2], [3,4,5])    
np.hstack([X.reshape(-1, 1), Y.reshape(-1, 1)])

#array([[1, 3],
#       [2, 3],
#       [1, 4],
#       [2, 4],
#       [1, 5],
#       [2, 5]])

@Denis提到的其他选项：

np.stack((X.ravel(), Y.ravel()), axis=-1)

至于速度，这两个选项具有可比性：

X, Y = np.meshgrid(np.arange(1000), np.arange(1000))

%timeit np.hstack([X.reshape(-1, 1), Y.reshape(-1, 1)])
#100 loops, best of 3: 4.77 ms per loop

%timeit np.stack((X.ravel(), Y.ravel()), axis=-1)
#100 loops, best of 3: 4.89 ms per loop