第一题
将np.arange(10)数组中的奇数全部都替换成-1。
代码如下(示例):
improt numpy as np
a = np.arange(10)
a[(a == 1)|(a == 3)|(a == 5)|(a == 7)|(a == 9)] = -1
print(a)
array([ 0, -1, 2, -1, 4, -1, 6, -1, 8, -1])
第二题
有一个4行4列的数组(比如:np.arange(16).reshape((4,4))),请将其中对角线的数取出来形成一个一维数组。提示(使用np.eye)。
代码如下(示例):
improt numpy as np
L = np.arange(16).reshape(4,4)
L1=[L[0,0],L[1,1],L[2,2],L[3,3]]
a = np.array(L1)
print(a)
array([ 0, 5, 10, 15])