python按值数组排序字典

Question

我有一个以数组为元素的字典。 说：

masterListShort = {'a': [5, 2, 1, 2], 'b': [7, 2, 4, 1], 'c': [2, 0, 1, 1]}

我想按值的第一个元素对该字典进行反向排序。 然后我想将我的输出写入一个制表符分隔的文件，如下所示：

<key>    <value1>    <value2>    <value3>    etc.

我将字典写入文件的当前代码如下所示：

# write the masterListShort to file
outFile2 = open('../masterListShort.tsv', 'w')
for item in sorted(masterListShort):
    tempStr = '\t'.join(map(str, masterListShort[item]))
    outFile2.write(str(item) + '\t' + tempStr + '\n')

outFile2.close()

此代码工作正常，只是不对列表进行排序。 我希望我的输出以制表符分隔的文件格式编写。 所以：

b    7    2    4    1
c    5    2    1    2
a    2    0    1    1

到目前为止，我已经找到了以下命令，并且想知道是否可以将它们应用到我的代码中：

import operator
sorted(myDict, key=operator.itemgetter(1))

Answer 1

您需要对 values 进行排序，然后在 first 索引上（所以 0 用于从零开始的索引），并告诉 sorted() 反转顺序：

import operator

sorted(myDict.values(), key=operator.itemgetter(0), reverse=True)

如果没有 dict.values() 调用，您将尝试对键进行排序。

演示：

>>> import operator
>>> myDict = {'a': [5, 2, 1, 2], 'b': [7, 2, 4, 1], 'c': [2, 0, 1, 1]}
>>> sorted(myDict.values(), key=operator.itemgetter(0), reverse=True)
[[7, 2, 4, 1], [5, 2, 1, 2], [2, 0, 1, 1]]

如果您想输出键值对，则使用dict.items() 并使用 lambda 访问仅值的第一个索引：

sorted(myDict.items(), key=lambda i: i[1][0], reverse=True)

演示：

>>> sorted(myDict.items(), key=lambda i: i[1][0], reverse=True)
[('b', [7, 2, 4, 1]), ('a', [5, 2, 1, 2]), ('c', [2, 0, 1, 1])]

在这两种情况下，实际上按第一个元素排序只是并没有太大意义；您可以让列表自然排序：

sorted(myDict.values(), reverse=True)  # sort just the values
sorted(myDict.items(), key=operator.itemgetter(1), reverse=True)  # sort items

因为列表的列表是按字典顺序排序的；如果第一个元素相等，则根据第二个元素对两个列表进行排序，依此类推。

要将其写入制表符分隔的文件，请使用csv 模块；它将负责将值转换为字符串、编写制表符分隔符和处理换行符：

import csv

with open('../masterListShort.tsv', 'wb') as outfh:
    writer = csv.writer(outfh, delimiter='\t')
    for key, values in sorted(myDict.items(), key=operator.itemgetter(1), reverse=True):
        writer.writerow([key] + values)

Answer 2

你可以对dict.items()给出的元组列表进行排序；

myDict = {'a': [5, 2, 1, 2], 'b': [7, 2, 4, 1], 'c': [2, 0, 1, 1]}

print myDict.items()
#[('a', [5, 2, 1, 2]), ('c', [2, 0, 1, 1]), ('b', [7, 2, 4, 1])]

print sorted(myDict.items(), key=lambda x: x[1][0], reverse=True))
#[('c', [2, 0, 1, 1]), ('a', [5, 2, 1, 2]), ('b', [7, 2, 4, 1])]

sorted 的 key 关键字参数是您提供给它的一个函数，它返回该函数可以用来对您给它进行排序的列表中的每个元素进行排序的东西。

lambda x: x[1][0] 只是一种快速创建函数的方法；

即

print sorted(myDict.items(), key=lambda x: x[1][0])

和

一样

def sortFunction(x):
  return x[1][0]

print sorted(myDict.items(), key=sortFunction)

它给出x[1][0] 的原因是列表 myDict.items() 是一个二维列表；每个元素都是一个元组，其中第一个元素是key，第二个元素是value。所以x[1] 是值（即分配给每个键的列表）。你想对它们的第一个值进行排序，所以我们对x[1][0]进行排序。

现在只需以您想要的格式打印它们；

for key, value in sorted(myDict.items(), key=sortFunction, reverse=True):
  print key + "\t" + "\t".join([str(v) for v in value])

这将给出：

b   7   2   4   1
a   5   2   1   2
c   2   0   1   1