NumPy如何改變數(shù)組的形狀

小億
86
2024-05-13 12:49:12

NumPy可以通過(guò)reshape()方法來(lái)改變數(shù)組的形狀。這個(gè)方法會(huì)返回一個(gè)新的具有指定形狀的數(shù)組，但是原始數(shù)組的數(shù)據(jù)不會(huì)被復(fù)制，而是通過(guò)改變數(shù)組的步長(zhǎng)來(lái)實(shí)現(xiàn)形狀的改變。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = arr.reshape((3, 2))

print(new_arr)

這將輸出:

[[1 2]
 [3 4]
 [5 6]]

在reshape()方法中，傳入一個(gè)元組作為參數(shù)，元組的每個(gè)元素表示新數(shù)組的每個(gè)維度的大小。需要注意的是，新數(shù)組的元素個(gè)數(shù)必須和原數(shù)組的元素個(gè)數(shù)一致。

0 踩