python编写一个数组求和函数Ad(a1, a2); a1是第一个数组， a2是第二个数组，函数返回两者之和result. 假设a1={2, 4, 5, 8}, a2={1, 0,4, 6},则sul=(3,4, 9, 14); 编写程序，依次输入n, a1, a2,输出result. n是数组的长度,两个数组的长度都为n。 第一行是数组长度n。 第二行是组成数组a1的n个整数。 第三行是组成数组a2的n个整数。 结果数组的n个元素。用一个空格隔开。

下面是一个示例代码，可以实现题目要求的功能：

def Ad(a1, a2):
    result = []
    for i in range(len(a1)):
        result.append(a1[i] + a2[i])
    return tuple(result)

n = int(input())
a1 = list(map(int, input().split()))
a2 = list(map(int, input().split()))

result = Ad(a1, a2)
print(*result)

输入示例：

4
2 4 5 8
1 0 4 6

输出示例：

3 4 9 14

相关问题

numba，定一个函数传入的是python普通的二维数组，实现按行求和

可以使用Numba的JIT编译器来加速Python函数。以下是一个使用Numba JIT编译器的示例代码，实现按行求和：

import numba as nb

@nb.jit(nopython=True)
def row_sum(arr):
    row_sums = []
    for row in arr:
        row_sum = 0
        for elem in row:
            row_sum += elem
        row_sums.append(row_sum)
    return row_sums

该函数使用了Numba的JIT编译器来加速计算。它接受一个二维数组作为输入，并返回一个包含每行和的列表。在函数中，我们遍历每一行并计算它们的和。然后将每行的和添加到一个列表中，并将该列表作为结果返回。

使用示例：

arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
row_sums = row_sum(arr)
print(row_sums)  # [6, 15, 24]

该示例将一个3x3的二维数组传递给`row_sum`函数，并打印函数的输出结果。输出结果为一个包含每行和的列表：[6, 15, 24]。

python中函数返回一个数组

在 Python 中，我们可以使用 NumPy 库创建数组，并在函数中返回数组。下面是一个简单的示例函数：

import numpy as np

def create_array():
    arr = np.array([1, 2, 3, 4, 5])
    return arr

在这个函数中，我们使用 NumPy 库创建了一个包含 1 到 5 的整数数组，并使用 `return` 语句返回该数组。

我们可以在主程序中调用这个函数，并将返回的数组赋值给一个变量：

my_array = create_array()
print(my_array)

输出结果为：

[1 2 3 4 5]

这样就成功地在函数中创建并返回了一个数组。当然，这只是一个简单的示例，实际中我们可能需要在函数中进行更复杂的数组操作，以满足具体的需求。