從列表的字典中返回包含每個鍵的前 N 個列表條目的字典

Question

考慮以下列表字典 d：

{
'ra': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'decl': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'source_id': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'priority': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
}

我想為每個字典鍵返回一個包含 <= 列表中前 N 個值的字典。 因此，如果N = 4 ，它應該返回

{
'ra': [0, 1, 2, 3],
'decl': [0, 1, 2, 3],
'source_id': [0, 1, 2, 3],
'priority': [0, 1, 2, 3],
}

或者如果列表少於 4 個條目，則返回完整列表。 有點像.head(N)適用於數據幀。

我可以將字典轉換為數據框，執行.head(N)操作並將其轉換回字典，但似乎必須有一種更簡單/更 Pythonic 的方法來做到這一點。

Answer 1

我會 go 使用字典理解的東西

lessen = {
'ra': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'decl': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'source_id': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
'priority': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
}
limit = 4
new = {k:v[:limit] for k,v in lessen.items()}

output

{'ra': [0, 1, 2, 3], 'decl': [0, 1, 2, 3], 'source_id': [0, 1, 2, 3], 'priority': [0, 1, 2, 3]}

Answer 2

使用簡單for循環很容易做到：

N = 4
for key in thedata:
    thedata[key] = thedata[key][:N]

Answer 3

您可以使用

dct = {
    'ra': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    'decl': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    'source_id': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
    'priority': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
}

new_dct = {key: values[0:4] for key, values in dct.items()}
print(new_dct)