Python List comprehension и JSON parsing

Question

Я новичок в Python и пытаюсь найти лучший способ проанализировать значения объекта JSON в массиве, используя понимание списка.

Вот мой код - я запрашивая общедоступную iNaturalist API, и хотел бы взять объект JSON, что он возвращается, так что я беру конкретные части объекта JSON в ухабистой массив:

import json 
import urllib2 

#Set Observations URL request for Resplendent Quetzal of Costa Rica 
query = urllib2.urlopen("http://api.inaturalist.org/v1/observations?place_id=6924&taxon_id=20856&per_page=200&order=desc&order_by=created_at") 
obSet = json.load(query) 

#Print out Lat Long of observation 
n = obSet['total_results'] 

for i in range(n) : 
    print obSet['results'][i]['location'] 

Это все работает отлично и дает следующий результат:

9.5142456535,-83.8011438905 
10.2335478381,-84.8517773638 
10.3358965682,-84.9964271008 
10.3744851815,-84.9871494128 
10.2468720343,-84.9298072822 
... 

То, что я хотел бы сделать дальше, заменить цикл со списком понимания и запоминания значения местоположения в кортеже. Я изо всех сил с синтаксисом в том, что я предполагаю, что это что-то вроде этого:

[(long,lat) for i in range(n) for (long,lat) in obSet['results'][i]['location']] 

Но это не работает ... спасибо за любую помощь.

Answer 1

obSet['results'] список, нет необходимости использовать range не итерацию над ней:

for item in obSet['results']: 
    print(item['location']) 

Чтобы сделать это в список понимания вы можете написать:

[item['location'] for item in obSet['results']] 

Но каждое местоположение кодируется строка, а не список или кортеж поплавков. Для того, чтобы получить его в нужный формат, используйте

[tuple(float(coord) for coord in item['location'].split(',')) 
for item in obSet['results']] 

То есть, разделить item['location'] строку на части, используя , в качестве разделителя, а затем преобразовать каждую часть в поплавок, и сделать кортеж из этих поплавка координат.

Answer 2

Вы можете перемещаться по списку результатов непосредственно:

print([tuple(result['location'].split(',')) for result in obSet['results']]) 
>> [('9.5142456535', '-83.8011438905'), ('10.2335478381', '-84.8517773638'), ... ] 

Answer 3

[tuple(obSet['results'][i]['location'].split(',')) for i in range(n)] 

Это возвращает список кортежей, элементы кортежей являются unicode.

Если вы хотите, чтобы элементы кортежей, как поплавки, выполните следующие действия:

[tuple(map(float,obSet['results'][i]['location'].split(','))) for i in range(n)] 

Answer 4

Другим способом получить список [долго, латы] без списка понимания:

In [14]: map(lambda x: obSet['results'][x]['location'].split(','), range(obSet['total_results'])) 
Out[14]: 
[[u'9.5142456535', u'-83.8011438905'], 
[u'10.2335478381', u'-84.8517773638'], 
[u'10.3358965682', u'-84.9964271008'], 
[u'10.3744851815', u'-84.9871494128'], 
... 

Если у вас хотел бы список кортежей вместо:

In [14]: map(lambda x: tuple(obSet['results'][x]['location'].split(',')), range(obSet['total_results'])) 
Out[14]: 
[[u'9.5142456535', u'-83.8011438905'], 
[u'10.2335478381', u'-84.8517773638'], 
[u'10.3358965682', u'-84.9964271008'], 
[u'10.3744851815', u'-84.9871494128'], 
... 

Если вы хотите преобразовать в поплавки тоже:

In [17]: map(lambda x: tuple(map(float, obSet['results'][x]['location'].split(','))), range(obSet['total_results'])) 
Out[17]: 
[(9.5142456535, -83.8011438905), 
(10.2335478381, -84.8517773638), 
(10.3358965682, -84.9964271008), 
(10.3744851815, -84.9871494128), 
(10.2468720343, -84.9298072822), 
(10.3456659939, -84.9451804822), 
... 

Answer 5

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

positions = [obSet['results'][i]['location'] for i in range(obSet['total_results'])] 

obSet['total_results'] является информативным, но не нужно, можно просто перебирает obSet['results'] непосредственно и использовать каждый полученный словарь:

positions = [res['location'] for res in obSet['results']] 

Теперь у вас есть список строк однако, поскольку каждый 'location' по-прежнему остается long,lat Отформатированная строка, которую вы напечатали ранее.

Split этой строки и конвертировать результат в последовательность поплавков:

positions = [map(float, res['location'].split(',')) for res in obSet['results']] 

Теперь у вас есть список списков со значениями с плавающей точкой:

>>> [map(float, res['location'].split(',')) for res in obSet['results']] 
[[9.5142456535, -83.8011438905], [10.2335478381, -84.8517773638], [10.3358965682, -84.9964271008], [10.3744851815, -84.9871494128], [10.2468720343, -84.9298072822], [10.3456659939, -84.9451804822], [10.3611732346, -84.9450302597], [10.3174360636, -84.8798676791], [10.325110706, -84.939710318], [9.4098152454, -83.9255607577], [9.4907141714, -83.9240819199], [9.562637289, -83.8170178428], [9.4373885911, -83.8312881263], [9.4766746409, -83.8120952573], [10.2651190176, -84.6360466565], [9.6572995298, -83.8322965118], [9.6997991784, -83.9076919066], [9.6811177044, -83.8487647156], [9.7416717045, -83.929327673], [9.4885099275, -83.9583968683], [10.1233252667, -84.5751029683], [9.4411815757, -83.824401543], [9.4202687169, -83.9550344212], [9.4620656621, -83.665183105], [9.5861809119, -83.8358881552], [9.4508914243, -83.9054016165], [9.4798058284, -83.9362558497], [9.5970449879, -83.8969131893], [9.5855562829, -83.8354434596], [10.2366179555, -84.854847472], [9.718459702, -83.8910277016], [9.4424384874, -83.8880459793], [9.5535916157, -83.9578166199], [10.4124554163, -84.9796942349], [10.0476688795, -84.298227929], [10.2129436252, -84.8384097435], [10.2052632717, -84.6053701877], [10.3835784147, -84.8677930134], [9.6079669672, -83.9084281155], [10.3583643315, -84.8069762134], [10.3975986735, -84.9196996767], [10.2060835381, -84.9698814407], [10.3322929317, -84.8805587129], [9.4756504472, -83.963818143], [10.3997876964, -84.9127311339], [10.1777433853, -84.0673088686], [10.3346128571, -84.9306278215], [9.5193346195, -83.9404786293], [9.421538224, -83.7689452093], [9.430427837, -83.9532672942], [10.3243212895, -84.9653175843], [10.021698503, -83.885674888]] 

Если необходимо иметь кортежи а не списков, добавить tuple() звонок:

positions = [tuple(map(float, res['location'].split(','))) 
      for res in obSet['results']] 

Последнее также гарантирует, что выражение работает в Python 3 (где map() возвращает итератор, а не список); вы бы в противном случае придется использовать вложенный список понимание:

# produce a list of lists in Python 3 
positions = [[float(p) for p in res['location'].split(',')] for res in obSet['results']] 

Answer 6

Чтобы исправить способ получить список кортежей используя списочные бы:

def to_tuple(coords_str): 
    return tuple(coords_str.split(',')) 

output_list = [to_tuple(obSet['results'][i]['location']) for i in range(obSet['total_results'])] 

Вы, конечно, можете заменить to_tuple() с лямбда-функции, я просто хотел сделать этот пример ясным. Кроме того, вы можете использовать map(), чтобы иметь кортеж с поплавками вместо строки: return tuple(map(float,coords_str.split(','))).

Answer 7

Давайте попробуем дать этому выстрел, начиная с только 1 место:

>>> (long, lat) = obSet['results'][0]['location']

Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
ValueError: too many values to unpack 

Хорошо, так что не работает, но почему? Это потому, что координаты долготы и широты - всего одна строка, поэтому вы не можете сразу ее распаковать в виде кортежа. Мы должны сначала разделить его на две разные строки.

>>> (long, lat) = obSet['results'][0]['location'].split(",")

Отсюда мы хотим перебрать весь набор результатов, которые мы знаем, индексируются от 0 до п. tuple(obSet['results'][i]['location'].split(",")) даст нам кортеж долготы, широту для результата при индексе i, так:
>>> [tuple(obSet['results'][i]['location'].split(",")) for i in range(n)]
должен предоставить нам набор кортежей, которые мы хотим.