bottleneck源码
时间: 2023-06-10 22:08:12 浏览: 134
Bottleneck是一个用于高效计算瓶颈特征的Python库。以下是Bottleneck的源代码示例:
```
import numpy as np
from bottleneck import bottleneck as bn
x = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print('input array:\n', x)
# 沿着第一个轴计算最小值
min_val = bn.nanmin(x, axis=0)
print('minimum values along axis 0:\n', min_val)
# 沿着第二个轴计算最大值
max_val = bn.nanmax(x, axis=1)
print('maximum values along axis 1:\n', max_val)
# 计算数组的中位数
median_val = bn.nanmedian(x)
print('median value of array:\n', median_val)
# 通过比较数组的每一行,找到最小值的索引
min_idx = bn.argnanmin(x, axis=1)
print('index of minimum value in each row:\n', min_idx)
```
在这个示例中,我们导入了NumPy和Bottleneck库,并使用NumPy创建了一个3x3的数组。我们使用Bottleneck库的`nanmin`、`nanmax`、`nanmedian`和`argnanmin`函数计算了数组的不同特征,并使用打印语句输出了结果。
阅读全文