Comment augmenter la dimension de numpy

Méthodes pour augmenter les dimensions dans numpy : 1. Utilisez "np.newaxis" pour augmenter les dimensions. "np.newaxis" est une valeur d'index spéciale utilisée pour insérer une nouvelle dimension à une position spécifiée. Vous pouvez utiliser np à la position correspondante. .newaxis pour augmenter la dimension ; 2. Utilisez "np.expand_dims()" pour augmenter la dimension. La fonction "np.expand_dims()" peut insérer une nouvelle dimension à la position spécifiée pour augmenter la dimension du tableau

Le système d'exploitation de ce tutoriel : système Windows 10, Python version 3.11.4, ordinateur Dell G3.

Dans NumPy, vous pouvez utiliser la fonction np.newaxis ou np.expand_dims() pour augmenter les dimensions d'un tableau. Les deux méthodes sont décrites en détail ci-dessous.

Utilisez np.newaxis pour ajouter des dimensions :

np.newaxis est une valeur d'index spéciale utilisée pour insérer une nouvelle dimension à la position spécifiée. Les dimensions peuvent être augmentées en utilisant np.newaxis à l'emplacement correspondant.

import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5])
print(a.shape)  # 输出 (5,)
# 在a的行维度上增加一个新维度
a_newaxis = a[np.newaxis, :]
print(a_newaxis.shape)  # 输出 (1, 5)
# 在a的列维度上增加一个新维度
a_newaxis = a[:, np.newaxis]
print(a_newaxis.shape)  # 输出 (5, 1)

Dans le code ci-dessus, np.newaxis est utilisé pour ajouter une nouvelle dimension à la dimension de ligne et à la dimension de colonne du tableau d'origine a.

Utilisez np.expand_dims() pour augmenter les dimensions :

La fonction np.expand_dims() peut insérer une nouvelle dimension à la position spécifiée pour augmenter la dimension du tableau.

import numpy as np
# 创建一个二维数组
b = np.array([[1, 2, 3], [4, 5, 6]])
print(b.shape)  # 输出 (2, 3)
# 在b的行维度上增加一个新维度
b_expand_dims = np.expand_dims(b, axis=0)
print(b_expand_dims.shape)  # 输出 (1, 2, 3)
# 在b的列维度上增加一个新维度
b_expand_dims = np.expand_dims(b, axis=1)
print(b_expand_dims.shape)  # 输出 (2, 1, 3)

Dans le code ci-dessus, la fonction np.expand_dims() est utilisée pour ajouter une nouvelle dimension à la dimension de ligne et à la dimension de colonne du tableau d'origine b, et le paramètre axis est utilisé pour spécifier dans quelle dimension se trouve la nouvelle dimension. à ajouter.

Que vous utilisiez np.newaxis ou np.expand_dims(), ce sont des fonctions d'outils utilisées pour augmenter les dimensions d'un tableau. Vous pouvez choisir la méthode à utiliser pour atteindre l'objectif d'augmenter les dimensions en fonction de besoins spécifiques.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!