创建一个 1 行 2 列的二维 NumPy 数组

Create a 2-dimensional NumPy array with 1 row and 2 columns

是否可以创建 1 行 2 列(行向量)的二维 NumPy 数组?

这就是我正在做的事情 (from the documentation),但我想知道是否可以一步(更简单)完成:

X_new2 = np.array([8.5,156])
X_new2 = X_new2[np.newaxis, :]

我也试过:

X_new2 = np.array([[8.5], [156]])

但这是返回一列。

试试这个:

y = np.expand_dims(x, axis=0)
print(y.shape)

您可以使用以下语法获得与示例中相同的结果:

X_new2 = np.array([[8.5,156]])

(注意额外的 [] 以使数组的形状正确。)