对于一个4*4的数组，如何提取对角线的数据并组成一维数组？

问题：有一个4行4列的数组（比如：np.random.randint(0,10,size=(4,4))），请将其中对角线的数取出来形成一个一维数组。提示（使用np.eye）。
思路：我只会一个一个数据定位，可以用for循环，然后创建一个空列表，再一个个用append() 添加进去。
如何用np.eye()解答，完全没有思路。请求有经验的tutor指导。谢谢！

P.S. 麻烦最好能留下代码，方便阅读

你好，请使用

b = np.diag(a)

b即是你需要的