Bagaimana untuk meningkatkan dimensi numpy

Kaedah untuk meningkatkan dimensi dalam numpy: 1. Gunakan "np.newaxis" untuk meningkatkan dimensi "np.newaxis" ialah nilai indeks khas yang digunakan untuk memasukkan dimensi baharu pada kedudukan yang ditetapkan .

Sistem pengendalian tutorial ini: sistem Windows 10, Python versi 3.11.4, komputer Dell G3.

Dalam NumPy, anda boleh menggunakan fungsi np.newaxis atau np.expand_dims() untuk meningkatkan dimensi tatasusunan. Kedua-dua kaedah diterangkan secara terperinci di bawah.

Gunakan np.newaxis untuk menambah dimensi:

np.newaxis ialah nilai indeks khas yang digunakan untuk memasukkan dimensi baharu pada kedudukan yang ditentukan. Dimensi boleh ditingkatkan dengan menggunakan np.newaxis di lokasi yang sepadan.

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)

Dalam kod di atas, np.newaxis digunakan untuk menambah dimensi baharu pada dimensi baris dan dimensi lajur tatasusunan asal a.

Gunakan np.expand_dims() untuk meningkatkan dimensi: Fungsi

np.expand_dims() boleh memasukkan dimensi baharu pada kedudukan yang ditentukan untuk meningkatkan dimensi tatasusunan.

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)

Dalam kod di atas, fungsi np.expand_dims() digunakan untuk menambah dimensi baharu pada dimensi baris dan dimensi lajur tatasusunan b asal, dan parameter paksi digunakan untuk menentukan di mana dimensi dimensi baharu itu berada. untuk ditambah.

Sama ada anda menggunakan np.newaxis atau np.expand_dims(), ia adalah fungsi alat yang digunakan untuk meningkatkan dimensi tatasusunan Anda boleh memilih kaedah yang hendak digunakan untuk mencapai tujuan meningkatkan dimensi mengikut keperluan tertentu.

Atas ialah kandungan terperinci Bagaimana untuk meningkatkan dimensi numpy. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!