我正在处理带有csv文件的代码,然后返回csv文件中每一行的数据点列表。列表将按日期和地点排序。
代码使列表成为我想要的样子,但在调用时不会返回列表。
第一个函数读取文件并生成数据点。第二个函数调用第一个,排序和(希望)返回数据
def CreateDataStructure(data):
allData=[]
with open(data,'r') as data:
dataRead=data.readlines()
for line in dataRead[1:]:
splitList=line.split(",")
dataPoint =[splitList[25],splitList[1],{splitList[19]:splitList[9]}]
allData.append(dataPoint)
sortedData=sorted(allData)
return sortedData
def compileData(filename,counter,sortedData):
if counter==0:
sortedData=CreateDataStructure(filename)
compileData(filename,1,sortedData)
else:
if counter<len(sortedData):
if sortedData[0][0]==sortedData[1][0] and sortedData[0][1]==sortedData[1][1]:#change these back
compDict=dict(list(sortedData[0][2].items())+list(sortedData[1][2].items()))
sortedData[0]=[sortedData[0][0],sortedData[0][1],compDict]
sortedData.pop(1)
compileData(filename,counter,sortedData)
counter=counter+1
else:
sortedData+=[sortedData.pop(0)]
counter=counter+1
compileData(filename,counter,sortedData)
else:
from itertools import groupby
for key, locationGroup in groupby(sortedData, lambda x: x[0]):
bigList=[]
smallList=[]
for date in locationGroup:
smallList.append(date)
bigList.append(smallList)
print bigList
return bigList
print compileData("fakeData.csv",0,[])
当我运行此代码时,它会打印我想要的内容(biglist,我在下面粘贴以防你想知道)但返回None(noneType对象)。 为什么返回和打印会提供两种不同的东西,我该如何解决??
[[['744701', '40974', {'Alkalinity': '234'}], ['744701', '41018', {'Alkalinity': '252'}], ['744701', '41058', {'Alkalinity': '270.53'}]], [['744701', '40974', {'Alkalinity': '234'}], ['744701', '41018', {'Alkalinity': '252'}], ['744701', '41058', {'Alkalinity': '270.53'}]], [['744701', '40974', {'Alkalinity': '234'}], ['744701', '41018', {'Alkalinity': '252'}], ['744701', '41058', {'Alkalinity': '270.53'}]]]
答案 0 :(得分:0)
您的compileData
函数有多个执行分支,这意味着它可以在多个点返回。但是,只有一个(counter !=0 and counter >= len(sortedData)
,即最后一个else
分支返回任何内容。
例如:
if counter==0:
sortedData=CreateDataStructure(filename)
compileData(filename,1,sortedData)
即使compileData
调用设法到达最后一个else
分支,即打印并返回结果的分支,实际上也没有对结果进行任何操作。这意味着在compileData(filename,1,sortedData)
完成后,它返回的数据将被丢弃并继续执行函数,直到它最终到达结尾并隐式返回None
。
这应解决 问题:
def compileData(filename,counter,sortedData):
if counter==0:
sortedData=CreateDataStructure(filename)
return compileData(filename,1,sortedData)
else:
if counter<len(sortedData):
if sortedData[0][0]==sortedData[1][0] and sortedData[0][1]==sortedData[1][1]:#change these back
compDict=dict(list(sortedData[0][2].items())+list(sortedData[1][2].items()))
sortedData[0]=[sortedData[0][0],sortedData[0][1],compDict]
sortedData.pop(1)
counter=counter+1
return compileData(filename,counter,sortedData)
else:
sortedData+=[sortedData.pop(0)]
counter=counter+1
return compileData(filename,counter,sortedData)
else:
from itertools import groupby
for key, locationGroup in groupby(sortedData, lambda x: x[0]):
bigList=[]
smallList=[]
for date in locationGroup:
smallList.append(date)
bigList.append(smallList)
print bigList
return bigList
您可以returning early使您的功能(主观地)更容易理解,而不是嵌套多个ifs
。您的功能将如下所示:
from itertools import groupby
...
def compileData(filename,counter,sortedData):
if counter==0:
sortedData=CreateDataStructure(filename)
return compileData(filename,1,sortedData)
if counter<len(sortedData):
if sortedData[0][0]==sortedData[1][0] and sortedData[0][1]==sortedData[1][1]:#change these back
compDict=dict(list(sortedData[0][2].items())+list(sortedData[1][2].items()))
sortedData[0]=[sortedData[0][0],sortedData[0][1],compDict]
sortedData.pop(1)
counter=counter+1
return compileData(filename,counter,sortedData)
sortedData+=[sortedData.pop(0)]
counter=counter+1
return compileData(filename,counter,sortedData)
for key, locationGroup in groupby(sortedData, lambda x: x[0]):
bigList=[]
smallList=[]
for date in locationGroup:
smallList.append(date)
bigList.append(smallList)
print bigList
return bigList