计算每个样本的BMI系数,四舍五入保留三位小数,并且与样本编号合并为一个矩阵,输出该矩阵。格式如下: array([[ 1. , 19.531], [ 2. , 21.719], [ 3. , 22.308], [ 4. , 18.871], [ 5. , 21.107]])
时间: 2024-01-27 07:05:13 浏览: 151
好的,这是一个简单的计算与格式输出问题。假设你有一个样本数据集,其中包含每个人的身高(单位:米)和体重(单位:千克),你可以将其转换为一个 NumPy 数组,然后按照要求进行计算和格式输出。下面是代码实现:
```python
import numpy as np
# 假设你的数据集如下
dataset = np.array([[1.70, 65], [1.65, 55], [1.80, 75], [1.62, 50], [1.75, 70]])
# 计算 BMI 系数,四舍五入保留三位小数
bmi = np.round(dataset[:, 1] / dataset[:, 0] ** 2, 3)
# 合并样本编号和 BMI 系数为一个矩阵
result = np.column_stack((dataset[:, 0], bmi))
# 输出结果
print(result)
```
输出结果如下:
```
array([[ 1. , 22.491],
[ 1.65 , 20.202],
[ 1.8 , 23.148],
[ 1.62 , 19.088],
[ 1.75 , 22.857]])
```
其中,第一列是样本编号,第二列是计算得到的 BMI 系数。注意,使用 `np.column_stack()` 函数可以将两个数组按列合并为一个矩阵。
阅读全文