2020.9.8携程笔试题目，数字按对角线输入到矩阵里。（python）

朋友做的携程笔试题目，完整题目记不得了，只能记得大概，用python写的。
题目描述：输入m，n。那么久有一个m*n的矩阵，m代表行，n代表列。现在要求你把1~n的数字按对角线依次填入矩阵。
示例：
m=7 n=6

结果：
[[ 1. 3. 6. 10. 15. 21.]
[ 2. 5. 9. 14. 20. 27.]
[ 4. 8. 13. 19. 26. 32.]
[ 7. 12. 18. 25. 31. 36.]
[11. 17. 24. 30. 35. 39.]
[16. 23. 29. 34. 38. 41.]
[22. 28. 33. 37. 40. 42.]]

import numpy as np
m,n = map(int,input().split