查找生成 200 个随机数的数组中最小 4 个值的平均值

Question

我在 spyder IDE 中运行了以下代码：idnum = 201034628seed(idnum);w = np.random.rand(200)print(w)这会生成以下结果：[0.00176212 0.79092217 0.1759531  0.00239256 0.78842458 0.30404404 0.25633004 0.88271124 0.72031936 0.17356416 0.5674158  0.83897948 0.4133943  0.22471237 0.66562002 0.70207085 0.55722598 0.86308392 0.14584968 0.66224337 0.79900625 0.2687224  0.45508786 0.99014178 0.176943   0.42335567 0.41034833 0.75497287 0.41301282 0.11294302 0.58715198 0.01524138 0.58633177 0.9784454  0.14610789 0.68654175 0.94733177 0.93776749 0.17294272 0.7491281  0.94087871 0.60510781 0.43708462 0.77303273 0.13250525 0.50794632 0.36706808 0.46873059 0.99757662 0.144249   0.69427544 0.78359245 0.64836852 0.16574067 0.98633778 0.05613428 0.51713291 0.27246708 0.26216551 0.44605373 0.99963659 0.90569603 0.31139955 0.25559081 0.8295379  0.84638476 0.48194161 0.505123   0.57456517 0.62727722 0.11940848 0.49435157 0.07438197 0.11481526 0.74184931 0.94697125 0.93788422 0.3586455 0.852594   0.35167897 0.57139446 0.77923007 0.09070311 0.07821641 0.38140649 0.80945136 0.81820638 0.8140444  0.94458644 0.42983398 0.06609377 0.25737315 0.27873234 0.87183073 0.14317078 0.8964766 0.00731705 0.16095917 0.70980283 0.49757526 0.06990482 0.15304861 0.02710815 0.21319381 0.82069776 0.19839614 0.64250566 0.6383788 0.12539173 0.74583486 0.11041236 0.827742   0.20340574 0.03643315 0.62638826 0.12454928 0.64567226 0.04782684 0.88455847 0.62114705 0.82253557 0.12590787 0.99624612 0.0780055  0.38312778 0.56969024 0.21771078 0.18022973 0.06825607 0.05189065 0.19410785 0.93458232 0.84006441 0.8796388  0.00574523 0.92213916 0.60108549 0.48774697 0.79918579 0.05700109 0.42167703 0.26358089 0.37023659 0.05556867 0.1788227  0.63840475 0.79772203 0.20969062 0.55459356 0.81425831 0.06324903 0.274849   0.15092814 0.65504038 0.57138257 0.37113864我需要从 w 数组中的数字中找到最小的 4 个值的平均值。我该怎么做？

Answer 1

您可以使用heapq.nsmallestwhich 应该比排序稍快：

import heapq

import statistics

print(statistics.mean(heapq.nsmallest(4, w)))

Answer 2

你可以简单地做：

sum(sorted(w)[:4])/len(sorted(w)[:4])

Answer 3

尝试对数组进行排序并通过切片选择 4 个最小值，然后找到平均值。

np.mean(np.sort(w)[:4])