«

Numpy的广播情况有哪些

时间:2024-4-4 22:08     作者:韩俊     分类: Python


本篇内容介绍了“Numpy的广播情况有哪些”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

Numpy的广播的三种情况

广播(Broadcast)是 numpy 对不同形状(shape)的数组进行数值计算的方式,当运算中的 2 个数组的形状不同时,numpy 将自动触发广播机制;即:可扩充较小数组中的元素来适配较大数组的形状,这种机制叫作广播(broadcasting) 广播机制如下,一共三种情况:由于日常应用中最常见的只有一维数组与二维数组,所以就不举三维及以上数组的例子了;这样更容易理解;

1. 有一个数组是一个数字,即可广播;

有一个数组是一个数字,则可以将该数字广播,如下图:

举例如下:

a = np.arange(3)
b = 5
print(a+b)
# 输出为:[5 6 7]

2.  维度的尾部一致,即可广播;

维度的尾部一致,即可广播;如下图,尾部维度均为3;

举例如下:尾部维度均为4;

a = np.arange(12).reshape(3,4)
b= np.arange(4)
print(a)   
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]
print(b)
# [0 1 2 3]
print(a+b)
# [[ 0  2  4  6]
#  [ 4  6  8 10]
#  [ 8 10 12 14]]

3. 两个数组均为一维数组,一个为行方向,一个为列方向,即可广播;

举例如下:

a = np.arange(3).reshape(3,1)
b= np.arange(3)
print(a)
# [[0]
#  [1]
#  [2]]
print(b)
# [0 1 2]
print(a+b)
# [[0 1 2]
#  [1 2 3]
#  [2 3 4]]

标签: python

热门推荐