數學建模例題2.27 陣列元素的索引示例

`import numpy as np
a = np.arange(16).reshape(4,4) #生成4行4列的陣列
b = a[1][2] #輸出6
c = a[1, 2] #同b
d = a[1:2, 2:3] #輸出[[6]]
x = np.array([0, 1, 2, 1])
print(a[x==1]) #輸出a的第2、4行元素

print("學號：3005")`