numpy数组矢量化

最新推荐文章于 2022-09-09 10:04:38 发布

老吴的私房菜

最新推荐文章于 2022-09-09 10:04:38 发布

阅读量2.3k

点赞数

分类专栏： Python与数据分析

本文链接：https://blog.csdn.net/laowu8615/article/details/62040133

版权

Python与数据分析专栏收录该内容

5 篇文章 0 订阅

订阅专栏

NumPy数组可以使用数组表达式来处理数据。用数组表达式打胎循环的做法，通常称为矢量化。一般来说，矢量化数组运算要比等价的Python方式快很多。假设我们想要在一组值上计算函数sqrt(x^2 + y^2)

In [1] : import numpy as np
points = np.arange(-3,3,0.01)
xs,ys = np.meshgrid(points,points)
ys

Out[1]:array([[-3.  , -3.  , -3.  , ..., -3.  , -3.  , -3.  ],       [-2.99, -2.99, -2.99, ..., -2.99, -2.99, -2.99],
       [-2.98, -2.98, -2.98, ..., -2.98, -2.98, -2.98],
       ..., 
       [ 2.97,  2.97,  2.97, ...,  2.97,  2.97,  2.97],
       [ 2.98,  2.98,  2.98, ...,  2.98,  2.98,  2.98],
       [ 2.99,  2.99,  2.99, ...,  2.99,  2.99,  2.99]])
对该数组进行求值，可以把这两个数组当做两个浮点数进行运算：
  
  
   
   
    
    
     
     
      
      
       
       
      
      
      
      
       
        
         
          
           
            
             
            
            
            
             
             
             
            
            
            
            import matplotlib.pyplot as plt 
            z = np.sqrt(xs**2 + ys**2) 
            z 
            
           
          
         
       
       
       
      
      
     
     
    
    
   
   
  
  
  
  
   
   
   
   
    
    
     
     
      
      Out[10]:
     
     
     
     
      
      array([[ 4.24264069,  4.23557552,  4.2285222 , ...,  4.22148078,
         4.2285222 ,  4.23557552],
       [ 4.23557552,  4.22849855,  4.22143341, ...,  4.21438014,
         4.22143341,  4.22849855],
       [ 4.2285222 ,  4.22143341,  4.21435642, ...,  4.20729129,
         4.21435642,  4.22143341],
       ..., 
       [ 4.22148078,  4.21438014,  4.20729129, ...,  4.20021428,
         4.20729129,  4.21438014],
       [ 4.2285222 ,  4.22143341,  4.21435642, ...,  4.20729129,
         4.21435642,  4.22143341],
       [ 4.23557552,  4.22849855,  4.22143341, ...,  4.21438014,
         4.22143341,  4.22849855]])