热线电话：13121318867

登录
首页大数据时代请问numpy中怎么删掉值为1的维度？
请问numpy中怎么删掉值为1的维度？
2023-04-28
收藏

在 NumPy 中，可以使用 np.squeeze() 函数来删除值为 1 的维度。本文将详细介绍 np.squeeze() 函数的用法和示例。

什么是 np.squeeze() 函数？

np.squeeze() 函数是 NumPy 库中的一个函数，用于从数组的形状中删除单维条目。如果将一个数组作为参数传递给 np.squeeze()，则该函数将返回一个新数组，其中所有维度为 1 的轴都被删除。此外，如果您希望只删除特定的维度，请在 np.squeeze() 函数的第二个参数中指定要删除的维度。

np.squeeze() 函数的语法

np.squeeze() 函数的语法如下所示：

numpy.squeeze(arr, axis=None)

其中：

  • arr：需要压缩的数组。
  • axis：默认为 None，表示删除所有维度为 1 的轴，也可以指定要删除的特定轴。

np.squeeze() 函数的示例

现在，让我们通过一些示例来了解 np.squeeze() 函数的使用。

示例 1：删除所有维度为 1 的轴

假设有一个形状为 (1, 2, 1, 3) 的数组。使用 np.squeeze() 函数可以从数组中删除维度为 1 的轴，并返回形状为 (2, 3) 的新数组。

import numpy as np

a = np.ones((1, 2, 1, 3))
b = np.squeeze(a)

print("Shape of a:", a.shape)
print("Shape of b:", b.shape)

输出：

Shape of a: (1, 2, 1, 3)
Shape of b: (2, 3)

示例 2：删除指定的维度

在这个例子中，我们将仅删除第二维和第四维。

import numpy as np

a = np.ones((1, 2, 3, 1, 4))
b = np.squeeze(a, axis=(1, 3))

print("Shape of a:", a.shape)
print("Shape of b:", b.shape)

输出：

Shape of a: (1, 2, 3, 1, 4)
Shape of b: (1, 3, 4)

总结

np.squeeze() 函数可以方便地删除数组中单维条目。您可以使用它来删除特定轴上的值为 1 的维度，也可以使用它来删除所有维度为 1 的轴。希望本文能够对您理解 np.squeeze() 函数有所帮助。


CDA数据分析师考试相关入口一览（建议收藏）：

▷ 想报名CDA认证考试，点击>>>   “CDA报名” 了解CDA考试详情；

▷ 想学习CDA考试教材，点击>>> “CDA教材” 了解CDA考试详情；

▷ 想加入CDA考试题库，点击>>> “CDA题库” 了解CDA考试详情；

▷ 想了解CDA考试含金量，点击>>> “CDA含金量” 了解CDA考试详情；

numpy 压缩

数据分析咨询请扫描二维码

若不方便扫码，搜微信号：CDAshujufenxi

上一篇pandas怎么读取每月一号的数据？
下一篇numpy如何计算任意底数的对数？
数据分析师考试动态
CDA报考指南
数据分析学习
数据分析师资讯
更多

CDA教育闭环

常见问题

关于我们

CDA数据分析师公众号

  • CDA数据分析师公众号

CDA考试中心小程序

  • CDA考试中心服务号

CDA数据分析师App下载

  • CDA数据分析师App下载
Copyright © 2015-2021, www.cda.cn All Rights Reserved. CDA数据分析师(北京国富如荷网络科技有限公司) 版权所有 京ICP备11001960号-9 京公网安备 11010802034615号 经营许可证编号：京B2-20210330
联系电话：13321103290 (微信同号)

OK
客服在线
立即咨询
客服在线
立即咨询