Получение списка определенных элементов индекса из списка словарей в Python (понимание списка)

Question

У меня есть список словарей примерно так:

listDict = [{'id':1,'other':2},{'id':3,'other':4},{'id':5,'other':6}]

Я хочу получить список всех идентификаторов из словарей. Итак, из данного списка я бы получил список:

[1,3,5]

Это должна быть одна строка. Я знаю, что делал это раньше, я просто забыл синтаксис ... Спасибо

Answer 1

>>> listDict = [{'id':1,'other':2},{'id':3,'other':4},{'id':5,'other':6}]
>>> [item["id"] for item in listDict]
[1, 3, 5]

Answer 2

Более концептуально приятный и потенциально более быстрый метод в зависимости от размера ваших данных.

Использование пакета pandas для простой ссылки на ключи как на заголовки столбцов и групповые значения с использованием одного и того же ключа:

import pandas as pd
listDict = [{'id':1,'other':2},{'id':3,'other':4},{'id':5,'other':6}]
df = pd.DataFrame(listDict)

# Then just reference the 'id' column to get a numpy array of it
df['id']

# or just get a list
df['id'].tolist()

Некоторые тесты ниже, панды явно превосходят большие данные. В небольшом кейсе используются 3 записи, в большом - 150 тыс. Записей:

setup_large = "listDict = [];\
[listDict.extend(({'id':1,'other':2},{'id':3,'other':4},\
{'id':5,'other':6})) for _ in range(50000)];\
from operator import itemgetter;import pandas as pd;\
df = pd.DataFrame(listDict);"

setup_small = "listDict = [];\
listDict.extend(({'id':1,'other':2},{'id':3,'other':4},{'id':5,'other':6}));\
from operator import itemgetter;import pandas as pd;\
df = pd.DataFrame(listDict);"

method1 = '[item["id"] for item in listDict]'
method2 = "df['id'].tolist()"

import timeit
t = timeit.Timer(method1, setup_small)
print('Small Method LC: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup_small)
print('Small Method Pandas: ' + str(t.timeit(100)))

t = timeit.Timer(method1, setup_large)
print('Large Method LC: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup_large)
print('Large Method Pandas: ' + str(t.timeit(100)))

#Small Method LC: 8.79764556885e-05
#Small Method Pandas: 0.00153517723083
#Large Method LC: 2.34853601456
#Large Method Pandas: 0.605192184448

Answer 3

[i['id'] for i in listDict]

Answer 4

Для фанатов Python:

import operator
map(operator.itemgetter('id'), listDict)

Answer 5

[elem ['id'] для элемента в listDict]