2016-08-28 107 views
0

我是Python的新手,并试图找出使用列表理解解析JSON对象值到数组中的最佳方法。Python列表理解和JSON解析

这里是我的代码 - 我查询公开可用的API iNaturalist并想借此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']] 

但这不工作......感谢任何帮助。

+1

该json中的位置是*字符串*。 –

+2

另外,由于你是Python新手,是否有任何特别的理由让你使用Python 2,现在Python 3已经出现了** 8年**? –

+0

呃,优点!我可能最好把它弄清楚。 –

回答

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']字符串转换成使用,作为分隔符的部分,然后每个部分转换为浮动,并且使这些浮动的元坐标。

2

可以遍历直接结果列表:

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

你正在返回一个字符串列表,而不是元组列表。 –

+0

'我接下来要做的是用列表理解替换for循环,并将位置值存储在一个元组中。“ - 这是OP的要求。 –

+0

@AhsanulHaque固定。 – DeepSpace

2
[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)] 
2

另一种方式来获得的[长,纬度]列表中没有列表理解:

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), 
... 
2

直接翻译你的代码到一个列表的理解是:

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之前打印的格式化字符串。

拆分该字符串,并将结果转换成浮动的序列:

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']] 
+0

感谢教授关于使用'map'的OP,当时我说服他尝试了Python 3;) –

+0

@AnttiHaapala:好了,'tuple()'调用将很好地排序出来;-) –

1

要纠正的方式来获得元组的列表应用列表解析是:

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()一个lambda函数,我只是想让这个例子清楚。此外,你可以使用map()有一个浮点数组而不是字符串:return tuple(map(float,coords_str.split(',')))

1

让我们尝试给这个一杆,从仅有1个位置:

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

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

好了,没有工作,但是为什么呢?这是因为经纬度坐标只有1个字符串,所以你不能立即将它作为一个元组解包。我们必须首先将它分成两个不同的字符串。

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

从这里,我们会希望通过一整套的结果,我们知道从0索引到n的迭代。 tuple(obSet['results'][i]['location'].split(","))会给我们在索引i的结果的经度,纬度的元组,所以:
>>> [tuple(obSet['results'][i]['location'].split(",")) for i in range(n)]
应该给我们一组我们想要的元组。