np.newaxis的作用

最新推荐文章于 2023-12-12 14:25:25 发布

原创最新推荐文章于 2023-12-12 14:25:25 发布 · 700 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#np.newaxis

python笔记专栏收录该内容

101 篇文章

订阅专栏

本文详细介绍如何使用NumPy库中的linspace函数生成等差数列，并通过[:,np.newaxis]语法将一维数组转换为二维数组的过程。展示了数组形状的变化及其在实际应用中的意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np

x_data = np.linspace(0, 10, 10) 
print("x_data形状:", x_data.shape, "\nx_data值：", x_data)
x_data2 = x_data[:, np.newaxis]  # 将一维数组，转换为二维数组
print("\nx_data2形状:", x_data2.shape, "\nx_data2值：", x_data2)

结果（注意形状的改变）

x_data形状: (10,) 
x_data值： [ 0.          1.11111111  2.22222222  3.33333333  4.44444444  5.55555556
  6.66666667  7.77777778  8.88888889 10.        ]

x_data2形状: (10, 1) 
x_data2值： [[ 0.        ]
 [ 1.11111111]
 [ 2.22222222]
 [ 3.33333333]
 [ 4.44444444]
 [ 5.55555556]
 [ 6.66666667]
 [ 7.77777778]
 [ 8.88888889]
 [10.        ]]