numpy数组计算

数组计算

查看数组形状

import numpy as np
t1=np.arange(3,12,2)
print(t1)#[ 3  5  7  9 11]
print(t1.shape)#(5,)

t2=np.array([[1,2,3],[4,5,6]])
print(t2)#[[1 2 3]
         # [4 5 6]]
print(t2.shape)#(2, 3)两行三列

t3=np.array([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]])
print(t3)#[[[ 1  2  3]
  #         [ 4  5  6]]

 #         [[ 7  8  9]
  #         [10 11 12]]]
print(t3.shape)#(2, 2, 3)

修改数组形状

t4=np.arange(12)
print(t4)#[ 0  1  2  3  4  5  6  7  8  9 10 11]

print(t4.reshape((3,4)))
"""[[ 0  1  2  3]
    [ 4  5  6  7]
    [ 8  9 10 11]]"""
print('-------------------')

三维数组变成二维一维

t5=np.arange(24).reshape((2,3,4))
print(t5)#[[[ 0  1  2  3]
   #        [ 4  5  6  7]
  #         [ 8  9 10 11]]

 #         [[12 13 14 15]
 #          [16 17 18 19]
 #          [20 21 22 23]]]
print(t5.reshape((4,6)))
"""[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]
 [12 13 14 15 16 17]
 [18 19 20 21 22 23]]"""
#赋值运算是将返回值返回到等式左端，与直接返回到显示器输出设备本质是相同的，只是承受的对象不同
print(t5.reshape((24,)))#[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
print(t5.reshape((24,1)))
"""[[ 0]
 [ 1]
 [ 2]
 [ 3]
 [ 4]
 [ 5]
 [ 6]
 [ 7]
 [ 8]
 [ 9]
 [10]
 [11]
 [12]
 [13]
 [14]
 [15]
 [16]
 [17]
 [18]
 [19]
 [20]
 [21]
 [22]
 [23]]"""
print(t5.reshape((1,24)))#[[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]]
print(t5.flatten())#数据转化为一维[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
print('-----------------------')

数组加减乘除

t6=np.arange(12)
print(t4)#[ 0  1  2  3  4  5  6  7  8  9 10 11]
print(t6)#[ 0  1  2  3  4  5  6  7  8  9 10 11]
print(t6+2)#[ 2  3  4  5  6  7  8  9 10 11 12 13]
print(t6+t4)#[ 0  2  4  6  8 10 12 14 16 18 20 22]
print(t6+t4-1)#[-1  1  3  5  7  9 11 13 15 17 19 21]
print(t6*2)#[ 0  2  4  6  8 10 12 14 16 18 20 22]
print(t6/2)#[0.  0.5 1.  1.5 2.  2.5 3.  3.5 4.  4.5 5.  5.5]
#print(t6/0)#[nan inf inf inf inf inf inf inf inf inf inf inf]
#   nan意思为not a number inf意为无穷大
print(t4*t6)#[  0   1   4   9  16  25  36  49  64  81 100 121]
print(t2)#[[1 2 3]
         # [4 5 6]]
print(t3)#[[[ 1  2  3]
  #         [ 4  5  6]]

 #         [[ 7  8  9]
  #         [10 11 12]]]
print(t2+t3)
'''[[[ 2  4  6]
  [ 8 10 12]]

 [[ 8 10 12]
  [14 16 18]]]'''

#广播原则，两个数组形状不相同时，也并不是完全不能进行计算，当其中一个数组形状为（a,b)时，另一个数组形状为（b)或者（a，1）时可以进行对应位置元素的数值运算，并且会广播到其他维数。
#331可以和任意（3,x）计算，反之除了（3,3），其余的（X,3）都不能计算
#再补充一下 补充一下:(a,b,c)可以和(b,c)以及(1.c) /(b,1)计算

当扩展到三维时，例如一个数组形状为（a,b,c)，那么当另一个数组的形状为（a,b,1),(b,c),©,(b,1)四种种情况时都可以完成广播计算。