我需要一个能够接受二维数组的函数
时间: 2024-09-07 13:06:40 浏览: 66
在Python中,如果你需要创建一个接受二维数组(也称为矩阵)作为输入的函数,你可以使用内置的numpy库,因为它提供了方便的操作二维数组的功能。以下是一个简单的示例,函数接收一个二维数组并返回该数组的一些统计信息:
```python
import numpy as np
def process_2d_array(arr):
"""
函数接受一个二维数组,并返回一些基本信息,如平均值、总和和形状。
参数:
arr (np.ndarray): 一个二维数组
返回:
dict: 包含平均值(mean)、总和(sum)和形状(shape)的信息
"""
result = {
'mean': np.mean(arr),
'sum': np.sum(arr),
'shape': arr.shape,
}
return result
# 示例
input_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
processed_info = process_2d_array(input_array)
阅读全文