speech02



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 [ ]: