使用 Python 查找文件的模式和中位数

Question

我在一个项目中的这部分代码遇到问题，并尝试以多种方式执行众数和中位数，但均未成功。但是，我确实需要在模式部分使用字典，所以那里的任何建议都会非常有帮助。

    # Find median
    order = converted_numbers.sort()
    middle = count/2
    if middle % 2 == 0:
        median = (converted_numbers[middle - 1] + converted_numbers[middle]) / 2
    else:
        median = converted_numbers[middle]

    # Mode calculations
    number_counts = {}
    mode = 0
    freq = 0
    for i in converted_numbers:
        if i in number_counts:
            number_counts[i] += 1
        else:
            number_counts[i] = 1
    for i in number_counts:
        counts = int(number_counts[i])
        mode = max(counts)

Answer 1

您的代码有一些问题：

• list.sort() 不返回值。因此，如果您想要一个排序列表（与原始列表分开）。

  ordered_numbers = converted_numbers[:] # copy it
  ordered_numbers.sort()
  

• 使用楼层划分，并检查count是否是偶数，而不是middle：

  middle = count // 2
  if count % 2 == 0:
      median = (ordered_numbers[middle - 1] + converted_numbers[middle]) / 2
  else:
      median = ordered_numbers[middle]
  

• 要计算模式，您可以使用 python 的核心库之一，collections 库，然后快速获取它：

  from collections import Counter
  
  counter = Counter(ordered_numbers) # no need to be sorted
  mode    = counter.most_common(1)   # returns the most commonly occuring item
  

编辑：

既然要求是使用字典，我们可以简单的写成这样：

number_counts = {}
for num in ordered_numbers:
    if number_counts.get(num, None):
        number_counts[num] = 1
    else:
        number_counts[num] += 1

mode = ordered_numbers[0] # set to a number already in list
mode_freq = 0
for num, freq in number_counts.items():
    if freq > mode_freq:
        mode, mode_freq = num, freq

您可以使用defaultdict 代替常规的dict，这样您就可以在元素存在时不需要

from collections import defaultdict

number_counts = defaultdict(int)
for num in ordered_numbers:
    number_counts[num] += 1

mode = #.... same as above