Numpy二維數組轉一維

一、Numpy一維數組轉二維

Numpy一維數組可以通過reshape函數轉換為二維數組。reshape函數的第一個參數是新數組的行數，第二個參數是列數。如果只給定一個參數，那麼reshape函數會將數組轉換為只有一列的二維數組。

import numpy as np

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

運行結果：

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

二、Numpy將二維數組轉為一維

Numpy提供了flatten和ravel函數，將二維數組轉換為一維數組。

import numpy as np

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

運行結果：

[1 2 3 4 5 6]

三、Numpy二維數組切片

Numpy支持對二維數組進行切片。切片的使用方式與對列表進行切片類似。

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
sub_arr = arr[1:, 1:]
print(sub_arr)

運行結果：

[[4]
 [6]]

四、Numpy三維數組賦值

Numpy中的三維數組可以通過賦值改變數組中的元素。

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
arr[0][0][0] = 0
print(arr)

運行結果：

[[[0 2]
  [3 4]]

 [[5 6]
  [7 8]]]

五、Numpy兩個一維數組合成二維數組

Numpy中的hstack和vstack函數可以將兩個一維數組沿著水平方向和垂直方向組合成二維數組。

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
new_arr = np.vstack((arr1, arr2))
print(new_arr)

運行結果：

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

六、Numpy二維數組排序

Numpy提供了sort函數和argsort函數對二維數組進行排序。sort函數將數組中的元素原地排序，argsort函數返回排序後元素的索引。

import numpy as np

arr = np.array([[1, 3], [0, 2]])
new_arr = np.sort(arr)
print(new_arr)

運行結果：

[[1 3]
 [0 2]]

七、Numpy三維數組理解

三維數組可以看作是二維的二維數組。

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr[0])

運行結果：

[[1 2]
 [3 4]]

八、一維數組怎麼變成二維數組numpy

可以使用reshape函數將一維數組轉換為二維數組。

import numpy as np

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

運行結果：

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

九、創建Numpy二維數組

可以使用array函數創建Numpy二維數組。

import numpy as np

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

運行結果：

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

十、Numpy數組不同維度轉換

Numpy提供了transpose函數、swapaxes函數和rollaxis函數對數組的維度進行轉換。

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
new_arr = np.transpose(arr, (1, 0, 2))
print(new_arr)

運行結果：

[[[1 2]
  [5 6]]

 [[3 4]
  [7 8]]]