Python:将嵌套列表对象写入csv文件
我正在尝试将列表中的数据写入csv文件。这是我所拥有的东西的简化版本Python:将嵌套列表对象写入csv文件,python,csv,nested-lists,Python,Csv,Nested Lists,我正在尝试将列表中的数据写入csv文件。这是我所拥有的东西的简化版本 class Point(object): def __init__(self, weight, height): self.weight = weight self.height = height def get_BMI(self): return (self.weight * self.height) / 42 # this is not how you c
class Point(object):
def __init__(self, weight, height):
self.weight = weight
self.height = height
def get_BMI(self):
return (self.weight * self.height) / 42 # this is not how you calculate BMI but let's say
myList = [[Point(30, 183)],[Point(63, 153)]]
由于数据的设置方式,我将点存储在嵌套循环中。如果我想访问第一个点对象的BMI,我会键入
myList[0][0].get_BMI()
我想将每个点的BMI写入CSV(用逗号分隔)。我该怎么做
我是这样想的,但这并不完全是直截了当的:
import csv
with open('output.csv', 'w') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(myList)
它不会返回任何错误,但实际上也不会创建CSV文件。我还想在myList[I][j]中写入值。将\u BMI()
获取到文件。我没有权限问题,因为我使用Spyder(python IDE)作为根。现在我只是通过Spyder控制台运行脚本,但它仍然可以工作并输出CSV文件。需要一个字符串或数字列表。您应该首先创建一个包含BMI值的列表,以便将它们写入您的csv
文件:
import csv
BMIs = [[point.get_BMI() for point in points] for points in myList]
with open('output.csv', 'w') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(BMIs)
有三个问题:
您可以使用
writerows
,因为它需要一个行的列表
——每个行
都应该按照方言
参数格式化为csv.writer
,在这种情况下可以忽略,而不必担心任何反作用
因此,writerows
可以采用类似于myList
的结构。问题是你需要访问所有的积分并获取他们的BMI(这可以在列表中完成)
为了说明如何使用writerows(并在每个点上添加一个数字,以便所有行都不只有一个条目(这将令人沮丧),我添加了enumerate调用
因此,您不再需要复杂的循环或任何类似的循环。享受:
myList = [[Point(30, 183)],[Point(63, 153)]]
with open('output.csv', 'w') as outfile:
writer = csv.writer(outfile)
writer.writerows(enumerate(p.get_BMI() for p in itertools.chain.from_iterable(myList)))
对于第3点,我认为
csv
正在默默地忽略既不是字符串也不是数字的对象,然后决定不写任何东西。@FrancisColasopen
调用本身将创建一个空文件,并且对象将有一个字符串表示形式,如csv
默认使用的
(通过str
运算符的等效项)。
myList = [[Point(30, 183)],[Point(63, 153)]]
with open('output.csv', 'w') as outfile:
writer = csv.writer(outfile)
writer.writerows(enumerate(p.get_BMI() for p in itertools.chain.from_iterable(myList)))
myList = [[Point(30, 183)],[Point(63, 153)]]
with open('output.csv', 'w') as outfile:
writer = csv.writer(outfile,delimiter=',')
for i in myList:
writer.writerow([i])