In [1]:

    

# 多行结果输出支持
from IPython.core.interactiveshell import InteractiveShell
InteractiveShell.ast_node_interactivity = "all"


    

In [2]:

    

import numpy as np


    

In [3]:

    

a = np.array([x + 2 for x in range(5)])


    

In [4]:

    

a


    

    
Out[4]:





array([2, 3, 4, 5, 6])

In [5]:

    

all(a)


    

    
Out[5]:





True

In [6]:

    

any(a)


    

    
Out[6]:





True

In [7]:

    

from itertools import accumulate


    

In [11]:

    

list(accumulate(a, max))


    

    
Out[11]:





[2, 3, 4, 5, 6]

In [12]:

    

b = np.diag(a)


    

In [13]:

    

b


    

    
Out[13]:





array([[2, 0, 0, 0, 0],
       [0, 3, 0, 0, 0],
       [0, 0, 4, 0, 0],
       [0, 0, 0, 5, 0],
       [0, 0, 0, 0, 6]])

In [14]:

    

c =np.diag(b)


    

In [15]:

    

c


    

    
Out[15]:





array([2, 3, 4, 5, 6])

In [ ]: