python创建一个创建2*2的数组，计算对角线上元素的和

可以使用Python中的NumPy库来创建2x2的数组，并且计算对角线上元素的和。下面是代码：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
print(arr)

# 计算对角线上元素的和
diagonal_sum = np.trace(arr)
print("对角线上元素的和为：", diagonal_sum)

输出结果为：

[[1 2]
 [3 4]]
对角线上元素的和为： 5

在以上代码中，我们使用`np.trace()`函数来计算数组的对角线上元素的和。该函数返回一个数组的对角线元素的总和。