在python中展平嵌套字符串列表

时间:2014-01-24 21:13:03

标签: python list csv nested flatten

我有两个列表,一个不是嵌套,另一个是。

carNames = []
priceAndMileage = []

数据看起来像这样,合并:

[(u'2011 Toyota Camry SE V6', [u'$14,995', u'31,750']), (u'2012 Toyota Camry L', [u'$15,993', u'27,381'])]

我的代码是:

combinedData = zip(carNames, priceAndMileage)

writer = csv.writer(open("dict.csv", 'r+'))

for dataList in combinedData:
    dataList = [dataList[0]] + [y for x in dataList[1] for y in x]
    writer.writerow(dataList)

我展平了dataList,但它遍历每个角色,而不仅仅是项目。如何生成诸如展平任务之类的结果并不会使每个字符变平,而只是将项目子列表展平?

我的结果是在csv文件中:

2011 Toyota Camry SE V6,$,1,4,",",9,9,5,3,1,",",7,5,0
2012 Toyota Camry L,$,1,5,",",9,9,3,2,7,",",3,8,1

但我需要:

2011 Toyota Camry SE V6, $14,995, 31,750
2012 Toyota Camry L, $15,993, 27,381

1 个答案:

答案 0 :(得分:3)

要展平dataList,您可以连接[dataList[0]]dataList[1]

   for dataList in combinedData:
        dataList = [dataList[0]] + dataList[1]
        writer.writerow(dataList)

<强>解释

dataList是,例如,(u'2011 Toyota Camry SE V6', [u'$14,995', u'31,750']) (顺便说一句,dataList不是list,它是tuple - 基本上是列表的不可变版本)

我们想得到的是扁平的dataList,即[u'2011 Toyota Camry SE V6', u'$14,995', u'31,750']

[dataList[0]]是一个只包含一个元素的列表:[u'2011 Toyota Camry SE V6']

dataList[1]是一个包含两个元素的列表:[u'$14,995', u'31,750']

[dataList[0]] + dataList[1]将连接这两个列表,我们将得到展平dataList