快捷導(dǎo)航

Numpy中np.newaxis的作用和用法小結(jié)

更新時間：2024年03月18日 14:21:52 作者：高斯小哥

np.newaxis常常用于將一個一維數(shù)組轉(zhuǎn)化為二維數(shù)組,本文就來介紹一下Numpy中np.newaxis的作用和用法小結(jié),具有一定的參考價值,感興趣的可以了解一下

一、np.newaxis的引入

在NumPy中，np.newaxis是一個非常有用的工具，它用于在數(shù)組的指定位置增加一個新的軸（維度）。理解并正確使用np.newaxis可以讓我們在處理多維數(shù)組時更加靈活，輕松實現(xiàn)數(shù)組的擴(kuò)展和重塑。

np.newaxis常常用于將一個一維數(shù)組轉(zhuǎn)化為二維數(shù)組，或者將二維數(shù)組轉(zhuǎn)化為三維數(shù)組等。它本質(zhì)上是一個占位符，表示我們想要在某個位置插入一個新的軸。

二、np.newaxis的基本用法

下面是一個簡單的例子，展示如何使用np.newaxis將一個一維數(shù)組轉(zhuǎn)化為二維數(shù)組：

import numpy as np

# 創(chuàng)建一個一維數(shù)組
arr_1d = np.array([1, 2, 3, 4])

# 使用np.newaxis增加一個新維度
arr_2d = arr_1d[np.newaxis, :]

print("一維數(shù)組:")
print(arr_1d)
print(arr_1d.shape)
print("轉(zhuǎn)化為二維數(shù)組:")
print(arr_2d)
print(arr_2d.shape)

輸出：

一維數(shù)組:
[1 2 3 4]
(4,)
轉(zhuǎn)化為二維數(shù)組:
[[1 2 3 4]]
(1, 4)

在這個例子中，arr_1d是一個一維數(shù)組，通過arr_1d[np.newaxis, :]，我們在數(shù)組的第一個維度（即最外層）增加了一個新的軸，將其轉(zhuǎn)化為一個二維數(shù)組。

三、np.newaxis的進(jìn)階用法

np.newaxis不僅可以在數(shù)組的第一個維度增加新軸，還可以在數(shù)組的任意位置增加新軸。下面是一個例子：

import numpy as np

# 創(chuàng)建一個二維數(shù)組
arr_2d = np.array([[1, 2], [3, 4]])

# 在數(shù)組的第二個維度（即列的方向）增加一個新軸
arr_3d = arr_2d[:, np.newaxis, :]

print("二維數(shù)組:")
print(arr_2d)
print(arr_2d.shape)
print("轉(zhuǎn)化為三維數(shù)組:")
print(arr_3d)
print(arr_3d.shape)

輸出：

二維數(shù)組:
[[1 2]
[3 4]]
(2, 2)
轉(zhuǎn)化為三維數(shù)組:
[[[1 2]]

[[3 4]]]
(2, 1, 2)

在這個例子中，arr_2d是一個二維數(shù)組，通過arr_2d[:, np.newaxis, :]，我們在數(shù)組的第二個維度增加了一個新的軸，將其轉(zhuǎn)化為一個三維數(shù)組。

四、np.newaxis與數(shù)組重塑

np.newaxis有時可以用來簡化數(shù)組的重塑操作。例如，我們有一個二維數(shù)組，想要將它重塑為一個三維數(shù)組，其中一個維度的大小為1，這時就可以使用np.newaxis。

import numpy as np

# 創(chuàng)建一個二維數(shù)組
arr_2d = np.array([[1, 2], [3, 4]])

# 使用reshape方法重塑為三維數(shù)組
arr_3d_reshape = arr_2d.reshape(arr_2d.shape[0], 1, arr_2d.shape[1])

# 使用np.newaxis簡化重塑操作
arr_3d_newaxis = arr_2d[:, np.newaxis, :]

print("使用reshape方法得到的三維數(shù)組形狀:")
print(arr_3d_reshape.shape)
print("使用np.newaxis得到的三維數(shù)組形狀:")
print(arr_3d_newaxis.shape)

輸出：