網站首頁 編程語言 正文
楔子
在 TensorFlow 中，可以給一個 tensor 增加一個維度、刪除一個維度，那么在 Numpy 中該怎么呢？
刪除維度、增加維度
先來看看如何增加一個維度：
import numpy as np
arr = np.array([[[1, 2, 3], [2, 3, 4]]])
print(arr)
"""
[[[1 2 3]
[2 3 4]]]
"""
print(arr.shape) # (1, 2, 3)
# 事實上第一個維度我們是不需要的，因為在該維度上數組的長度是 1
# 刪除第 1 個維度，我們看到已經改變了
print(np.squeeze(arr, 0))
"""
[[1 2 3]
[2 3 4]]
"""
但是注意：只有數組長度在該維度上為 1，那么該維度才可以被刪除。如果不是1，那么刪除的話會報錯。
import numpy as np
arr = np.array([[[1, 2, 3], [2, 3, 4]]])
print(arr.shape) # (1, 2, 3)
try:
# 刪除第二個維度，顯然在第二個維度上數組的長度是 2，不是 1
# 所以它不能被刪除
print(np.squeeze(arr, 1))
except Exception as e:
print(e) # cannot select an axis to squeeze out which has size not equal to one
刪除只能刪除數組長度為 1 所對應的維度，同理添加也是添加一個維度也只是讓數組在這個維度上的長度變成 1，因為數組本來不存在這個維度的，但是我們強行加上了一個維度，那么數組在這個維度上的長度只能是 1。
import numpy as np
arr = np.array([[1, 2, 3], [2, 3, 4]])
print(arr.shape) # (2, 3)
# 很好理解
print(np.expand_dims(arr, 0).shape) # (1, 2, 3)
print(np.expand_dims(arr, 1).shape) # (2, 1, 3)
print(np.expand_dims(arr, 2).shape) # (2, 3, 1)
arr = np.array([1, 2, 3])
print(np.expand_dims(arr, 0))
"""
[[1 2 3]]
"""
print(np.expand_dims(arr, 1))
"""
[[1]
[2]
[3]]
"""
以上就實現了數組維度的刪除和增加，因為數組的元素是固定的，所以在刪除維度和增加維度時，數組在該維度上的長度必須是 1。
另外，變化維度還可以使用 reshape，比如 arr 的維度是 (2, 1, 3)，我們把第二個維度給去掉的話，那么直接 arr.reshape((2, 3)) 即可，增加維度也是同理，只要變化維度前后的元素個數不變即可。
import numpy as np
arr = np.array([[1, 2, 3], [2, 3, 4]])
print(arr.shape) # (2, 3)
arr1 = arr.reshape((2, 1, 1, 3))
print(arr1)
"""
[[[[1 2 3]]]
[[[2 3 4]]]]
"""
print(arr1.shape) # (2, 1, 1, 3)
print(np.all(arr1.reshape((2, 3)) == arr)) # True
最后，增加維度還有一種做法，但用的不多，舉個栗子：
import numpy as np
arr = np.array([[1, 2, 3], [2, 3, 4]])
print(arr.shape) # (2, 3)
# 將維度變成 (2, 1, 3, 1, 1)
arr1 = arr[:, np.newaxis, :, np.newaxis, np.newaxis]
print(arr1.shape) # (2, 1, 3, 1, 1)
# np.newaxis 等價于 None
print(arr[:, None, :, None, None].shape) # (2, 1, 3, 1, 1)
# 使用 : 的部分和之前的維度是對應的，np.newaxis 或者 None 可以理解成 1
# 因此最終得到的數組的維度就是 (2, 1, 3, 1, 1)
# 再以一維數組為例
arr = np.array([1, 2, 3])
print(arr)
"""
[1 2 3]
"""
# 得到的數組的 shape 為 (1, 3)
print(arr[None, :])
"""
[[1 2 3]]
"""
# 得到的數組的 shape 為 (3, 1)
print(arr[:, None])
"""
[[1]
[2]
[3]]
"""
刪除、增加一行或一列
說實話，改變數組的維度不是特別常見，更常見的是刪除數組的一行或者一列，舉個栗子：
# 原始數組
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
# 我們希望刪除一行
[[ 0 1 2 3]
[ 8 9 10 11]]
# 或者刪除一列
[[ 0 2 3]
[ 4 6 7]
[ 8 10 11]]
這種需求相對來說更加常見一些，那么應該怎么做呢？我們來看一下。
刪除一行或一列
首先是刪除：
import numpy as np
arr = np.arange(0, 12).reshape(3, 4)
print(arr)
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 假設刪除第二行
print(np.delete(arr, [1], axis=0))
"""
[[ 0 1 2 3]
[ 8 9 10 11]]
"""
# 刪除第一行和第三行
print(np.delete(arr, [0, 2], axis=0))
"""
[[4 5 6 7]]
"""
# 刪除前兩行，slice(0, 2) 也可以換成 np.s_[0: 2]
print(np.delete(arr, slice(0, 2), axis=0))
"""
[[ 8 9 10 11]]
"""
刪除列的話也是同理，只需要將 axis=0 換成 axis=1 即可，注意：如果不指定 axis 或者 axis 指定為 None，那么會 np.delete 會將傳遞的數組扁平化（變成一維數組），然后進行刪除。舉個栗子：
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr)
"""
[[1 2 3]
[4 5 6]]
"""
# 會將 arr 扁平化處理，然后刪除索引為 1 的元素，因此要注意 axis 參數
print(np.delete(arr, [1]))
"""
[1 3 4 5 6]
"""
增加一行或一列
如果想增加一行或一列的話，要怎么做呢？
import numpy as np
arr = np.arange(0, 12).reshape(3, 4)
print(arr)
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 在尾部增加一行，注意：這里的維度一定要匹配，指定 [0, 0, 0, 0] 是不行的，因為 arr 是一個二維數組
print(np.append(arr, [[0, 0, 0, 0]], axis=0))
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[ 0 0 0 0]]
"""
# 在尾部增加一列，維度同樣要匹配
print(np.append(arr, [[0], [0], [0]], axis=1))
"""
[[ 0 1 2 3 0]
[ 4 5 6 7 0]
[ 8 9 10 11 0]]
"""
如果不指定 axis，那么仍然會將傳遞的數組扁平化，然后進行追加：
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr)
print(np.append(arr, 0)) # [1 2 3 4 5 6 0]
print(np.append(arr, [0, 0])) # [1 2 3 4 5 6 0 0]
print(np.append(arr, [[0, 0]])) # [1 2 3 4 5 6 0 0]
append 默認是在尾部進行追加，并且還要求維度要匹配，不是很方便。所以這里更推薦 insert 函數：
import numpy as np
arr = np.arange(0, 12).reshape(3, 4)
print(arr)
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 索引為 1 的位置插入一行，值全為 0
print(np.insert(arr, 1, 0, axis=0))
"""
[[ 0 1 2 3]
[ 0 0 0 0]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 或者我們也可以手動指定
print(np.insert(arr, 1, [0, 0, 0, 0], axis=0))
"""
[[ 0 1 2 3]
[ 0 0 0 0]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 二維數組也是可以的
print(np.insert(arr, 1, [[0, 0, 0, 0], [0, 0, 0, 0]], axis=0))
"""
[[ 0 1 2 3]
[ 0 0 0 0]
[ 0 0 0 0]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 插入一列，注意元素個數要匹配，每一列是 3 個元素
print(np.insert(arr, 1, [[0, 0, 0], [0, 0, 0]], axis=1))
"""
[[ 0 0 0 1 2 3]
[ 4 0 0 5 6 7]
[ 8 0 0 9 10 11]]
"""
我們看到 insert 比 append 要方便很多，并且功能也更加強大一些，并且 append 完全可以使用 insert 實現，舉個栗子：
import numpy as np
arr = np.arange(0, 12).reshape(3, 4)
print(arr)
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
# 在尾部增加一行
print(np.insert(arr, arr.shape[0], 0, axis=0))
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[ 0 0 0 0]]
"""
# 在尾部增加一列
print(np.insert(arr, arr.shape[1], 0, axis=1))
"""
[[ 0 1 2 3 0]
[ 4 5 6 7 0]
[ 8 9 10 11 0]]
"""
最后，如果 insert 不指定維度，那么也是會先將數組扁平化，然后在進行 insert，舉個栗子：
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(np.insert(arr, 1, [0, 0])) # [1 0 0 2 3 4 5 6]
總的來說還是比較簡單的。
原文鏈接：https://www.cnblogs.com/traditional/p/12629050.html
- 上一篇：沒有了
- 下一篇：沒有了
相關推薦
- 2022-03-24 Android繪制平移動畫的示例代碼_Android
- 2023-04-06 C++中的memset用法詳解_C 語言
- 2022-11-08 服務器nginx權限被拒絕解決案例_nginx
- 2022-04-08 WPF基本控件介紹_基礎應用
- 2024-01-15 mybatis中@Results,@ResultMap注解使用
- 2023-07-05 go gorm想要查詢數據按照where in中的數據進行排序
- 2022-11-02 Golang實現多存儲驅動設計SDK案例_Golang
- 2023-07-09 Python Django 零基礎從零到一部署服務，Hello Django！全文件夾目錄和核心代碼
- 欄目分類
-
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版（
- Linux 中運行的top命令 怎么退出？
- MySQL 中decimal 的用法？ 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符，流程控制 Flo
- 1. Int 和Integer 的區別，Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤：Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密？
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串（SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列，發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支
提供CDN加速